1289 大鱼吃小鱼（栈维护）

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