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
用栈结构,
如果向右,则入栈。
如果鱼向左游,则将此鱼与栈顶元素比较,如果大于站顶元素,出栈,否则被吃掉。
如果最后栈内元素个数为0,则此鱼会活下来。
还有最后别忘了,在栈里的元素。
#include<iostream>
#include<stack>
using namespace std;
stack <int > s;
int main()
{
int n,sum=0,num,dir;
cin>>n;
for (int i=0;i<n;i++)
{
cin>>num>>dir;
if(!dir)
{
while(s.size())
{
int t=s.top();
if(t>num)
break;
else
s.pop();
}
if (!s.size())
sum++;
}
else
s.push(num);
}
cout<<sum+s.size();
return 0;
}