1580-大鱼吃小鱼（栈）

题目来源：http://acm.nyist.cf/problem/1580

题目描述:

有N条鱼，在一条直线上移动。速度相同，两条鱼相遇大鱼会吃掉小鱼。

给出每条鱼的初始位置，大小，移动方向

时间足够长，最后还剩多少鱼。

所有鱼大小初始位置均不相同，位置5相对与位置2，5在2的右边。

 

输入描述:

多组输入，输入到文件结束，数据不超过100组。

每组数据第一行一个整数n（1 <= n <= 1000），表示鱼的数量

第二行n个整数用空格分隔，表示x_i（-10^9 <= x_i <= 10^9）

第三行n个整数用空格分隔，表示a_i（-10^9 <= a_i <= 10^9）

第四行n个整数用空格分隔，表示第i条鱼的方向，0表示向左，1表示向右

输出描述:

每组数据输出一个整数占一行表示答案。

样例输入:

复制

2
2 1
10 9
1 0
2
1 2
5 2
1 0

样例输出:

2
1

提示:

 

来源:

上传者：ToRe

思路：简单数据结构栈的处理（由于比赛时候把样例看错了导致没写出来）把鱼按位置从小到大排序，遇到朝右游入栈，朝左游就循环和栈顶比较，如果栈顶空了答案+1，最后答案再加上栈内剩余数据数量。

参考代码：

#include <iostream>
#include <cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<stack>
#include <map>
#include<set>
#include<cstdio>
#include<vector>
#include<list>
#include<ctime>
#include<bitset>
using namespace std;
const double pi = acos(-1);
#define N 1005
struct node
{
    int a,b,c;
} ans[N];
bool cmp(node x,node y)
{
    return x.a<y.a;
}
int n;
int main()
{
    while(cin>>n)
    {
        for(int i=0; i<n; i++) cin>>ans[i].a;
        for(int i=0; i<n; i++) cin>>ans[i].b;
        for(int i=0; i<n; i++) cin>>ans[i].c;
        sort(ans,ans+n,cmp);
        int num=0;
        stack<int>sum;
        for(int i=0; i<n; i++)
        {
            if(ans[i].c==0)
            {
                while(!sum.empty())
                {
                    if(ans[i].b>sum.top())
                        sum.pop();
                    else
                        break;
                }
                if(sum.empty())
                    ++num;
            }
            else
                sum.push(ans[i].b);
        }
        while(!sum.empty())
            sum.pop(),++num;
        cout<<num<<endl;
    }
    return 0;
}