题目来源: 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
思路: 用栈模拟整个过程, 当方向向右就入栈,向左,就和栈尾元素相比较,大于栈尾元素,cnt--, tail--, 直到tail==front,则 cnt++。若小于等于栈尾元素,则被吃掉,不需要处理。
( 千万不要误以为,大鱼吃掉小鱼后,大小会增加)
#include<stdio.h>
struct STACK{
int a,b;
} stack[100001];
int main(void) {
int i,N,front,tail,cnt=0,a,b;
scanf("%d",&N);
front = tail = 0;
while(N--){
scanf("%d%d",&a,&b);
if(b==1){ //方向向右,就入栈
stack[++tail].a = a;
stack[tail].b = b;
cnt++;
}
else {
while( tail > front) {
if( stack[tail].a < a ){
cnt--;
tail--;
}
else break;
}
if( tail == front && b==0 )
cnt++;
}
}
printf("%d\n",cnt);
return 0;
}