有N条鱼每条鱼的位置及大小均不同,他们沿着X轴游动,有的向左,有的向右。游动的速度是一样的,两条鱼相遇大鱼会吃掉小鱼。从左到右给出每条鱼的大小和游动的方向(0表示向左,1表示向右)。问足够长的时间之后,能剩下多少条鱼?
Input
第1行:1个数N,表示鱼的数量(1 <= N <= 100000)。
第2 - N + 1行:每行两个数Aii, Bii,中间用空格分隔,分别表示鱼的大小及游动的方向(1 <= Aii <= 10^9,Bii = 0 或 1,0表示向左,1表示向右)。
Output
输出1个数,表示最终剩下的鱼的数量。
Sample Input
5
4 0
3 1
2 0
1 0
5 0
Sample Output
2
一开始用数组,用向左的鱼往右边遍历,再用向右的往左边遍历,样例全过,就是Wrong Answer.也找不出bug,而且边界条件比较容易出错(喜欢从数学的角度看题,很多时候会想太多).
参考了各路大神的codes,原来用栈来模拟,那么perfect.
#include <bits/stdc++.h>
using namespace std;
stack<int> s;
int main(){
int n,ans;
int dir,size; //direction(方向),size(大小)
scanf("%d",&n);
ans=n; //一开始一共n条鱼
for(int i=1;i<=n;i++){
scanf("%d%d",&size,&dir);
if(dir) //根据题意,dir:1,向右.0,向左。
s.push(size); //鱼的方向向右,入栈。
else {
while(!s.empty()) { //判断一下栈的状态,非空说明里面之前有往右的鱼
if(size>s.top()){ //这里是往左的鱼的大小
s.pop(); //大鱼(向左)吃小鱼(向右)
ans--; //向右的鱼的数量-1
} else {
ans--; //向左的鱼的数量-1
break;
}
}
}
}
printf("%d\n",ans);
return 0;
}