点击打开题目
1289 大鱼吃小鱼
题目来源: Codility
基准时间限制:1 秒 空间限制:131072 KB 分值: 5 难度:1级算法题
收藏
关注
有N条鱼每条鱼的位置及大小均不同,他们沿着X轴游动,有的向左,有的向右。游动的速度是一样的,两条鱼相遇大鱼会吃掉小鱼。从左到右给出每条鱼的大小和游动的方向(0表示向左,1表示向右)。问足够长的时间之后,能剩下多少条鱼?
Input
第1行:1个数N,表示鱼的数量(1 <= N <= 100000)。
第2 - N + 1行:每行两个数A[i], B[i],中间用空格分隔,分别表示鱼的大小及游动的方向(1 <= A[i] <= 10^9,B[i] = 0 或 1,0表示向左,1表示向右)。
Output
输出1个数,表示最终剩下的鱼的数量。
Input示例
5
4 0
3 1
2 0
1 0
5 0
Output示例
2
从前往后记录的时候,前面的向左游的不用考虑,只有向左游而且存在向右的鱼的时候才开始对比。
用一个栈记录向右游动的鱼的体型,然后比较就行了。
注意,大鱼吃完小鱼后体型不增加。
代码如下:
#include <stdio.h>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
int main()
{
int n;
scanf ("%d",&n);
int ans = n;
stack<int> s;
while (n--)
{
int m,p;
scanf ("%d %d",&m,&p);
if (p == 0)
{
if (s.empty())
continue;
else
{
while (!s.empty())
{
if (m < s.top())
{
ans--;
break;
}
ans--;
s.pop();
}
}
}
else
s.push(m);
}
printf ("%d\n",ans);
return 0;
}