【51Nod】1289 - 大鱼吃小鱼（栈）

点击打开题目

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 
#include 
#include  
#include 
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 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;
}