有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;
}