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


阅读更多

没有更多推荐了,返回首页