大一积分赛 (一)——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

这道题的思路不难，就是两种鱼类的互吃有卡了很久，这道题最开始是用优先队列写的，分别用两种队列储存两种

不同游向的鱼，然后依次出队列，判断吃不吃掉队放的鱼，被吃掉的鱼出队知道任意一个队列为空，然后在或去两个

队列所剩元素个数之和。思路应该没问题，结果陷入死循环没输出，并且时间超限了

priority_queue<ll,vector<ll>,greater<ll> > q1;//向左游  0
		priority_queue<ll,vector<ll>,greater<ll> > q2;//向右游  1
		for(i=0;i<t;i++){
			scanf("%lld %lld",&a[i],&b[i]);
			if(b[i]==0) q1.push(a[i]);
			if(b[i]==1) q2.push(a[i]);
		}
		while(!q1.empty()||!q2.empty()){
			if(q1.top()<q2.top()) q1.pop();
			else if(q1.top()>q2.top()) q2.pop();
		}
		ll ans=q1.size()+q2.size();
		printf("%lld\n",ans);

然后就搜了一下题解，发现只用栈就可以实现这个操作了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
typedef long long ll;
int main(){
	int n;
	ll size,dir;
	while(cin>>n){
		stack<ll> s;
		int ans=0;
		for(int i=0;i<n;i++){
			scanf("%lld %lld",&size,&dir);
			if(dir) s.push(size);
			else{
				while(!s.empty()&&s.top()<size){
					s.pop();
					ans++;//如果向左游的鱼吃了向右游的鱼则ans++，表示少的鱼的数量 
				}
				if(!s.empty()) ans++;//同理 
			}
		}
		printf("%d\n",n-ans);
	}
	return 0;
}