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 <iostream>
using namespace std;
int a[100010] = {0};
int main()
{
	int abg = 0, aed = 1;//abg 存鱼开始的地方 aed存鱼结束的地方 实时更新
	int n, ta, tb;
	int ans = 0;
	cin>>n;
	for(int i = 0; i < n ; i++)
	{
		cin>>ta>>tb;
		if(tb == 1)
		{
			a[aed ++] = ta;//往右游存起来
		}
		else//往左游去吃往右游的
		{
			int tf = 1;//记录有没有全吃完
			for(int j = aed - 1; j > abg; j--)
			{
				if(a[j] < ta)
				{
					a[j] = 0;//吃了
					aed = j;//往右的少一个位置
				}
				else
				{
					tf = 0;//往左的吃不动了 凉了
					break;
				}
			}
			if(tf)//往左全吃完了 它活了
				ans ++;
		}
	}
	for(int i = 0; i < aed + 10; i++)
	{
		if(a[i] != 0)//记录还剩几条没被吃
			ans ++;
	}
	cout<<ans<<endl;
	return 0;
}

方法二： 栈操作

原理与上相同，但是代码篇幅小，易于实现

#include <iostream>
#include <stack>
using namespace std;
stack <int> stk;
int main()
{
	ios::sync_with_stdio(false);
	int n;
	int ans = 0;
	cin>>n;
	int big, mode;
	while(n -- )
	{
		
		cin>>big>>mode;
		if(mode == 0) //吃栈
		{
			while(!stk.empty() && big > stk.top())
			{
				stk.pop();
			}
			if(stk.empty())
			{
				ans ++;
			}
		}
		else //压栈
		{
			stk.push(big);
		}
	}
	cout<<stk.size() + ans<<endl;
	return 0;