题目来源: 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
一开始看到题目有点棘手,没啥思路,用数组模拟写不下去,用结构体也没写出来,然后就用栈来维护了,
就是将所有向右的鱼都存在栈里面,刚好就用到了栈的特点,然后只要每次遇到向左的就遍历一下栈就可以.
1 #include <bits/stdc++.h> 2 #define ll long long int 3 #define N 100005 4 using namespace std; 5 stack<int> s; 6 int main(){ 7 int n; 8 scanf("%d",&n); 9 int ans=n; 10 for(int i=0;i<n;i++){ 11 int x,y; 12 scanf("%d%d",&x,&y); 13 if(y==1) 14 s.push(x); 15 else{ 16 while(!s.empty()){ 17 if(s.top()<x){ 18 s.pop(); 19 ans--; 20 }else{ 21 ans--; 22 break; 23 } 24 } 25 } 26 } 27 printf("%d ",ans); 28 return 0; 29 }