51nodoj (大鱼吃小鱼---栈)

来源:互联网 发布:java开发职业规划 编辑:程序博客网 时间:2024/06/07 01:05

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

运用栈的知识,分出两个栈,用来存放所有的鱼和向左的鱼,然后进行比较即可

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#include <queue>#include <stack>using namespace std;#define M 220#define INF 0x3f3f3f#define RCL(a, b) memset(a, b, sizeof(a))struct node{    int size, direct;};node no, ne;int main(){    int n;    scanf("%d", &n);    stack<node> s1, s2;    for(int i=0; i<n; i++)    {        scanf("%d%d", &no.size, &no.direct);        s1.push(no);    }    int num = 0, ok = 0;    while(!s1.empty())    {        no = s1.top();//取出最右边的鱼看看         if(s2.empty())//判断最右边往左边走的鱼有没有         {            if(no.direct)//最右边的鱼向右边走的时候             {                num++;                s1.pop();            }            else            {                s2.push(no);//压入向最左边走的鱼的栈                 s1.pop();            }        }        else        {            if(!no.direct)//最右边的鱼是不是往左边走             {                s2.push(no);                s1.pop();            }            else            {                ne = s2.top();                if(ne.size < no.size)                {                    s2.pop();//吃掉往左边走的鱼                 }                else                {                    s1.pop();//被吃                 }            }        }    }    while(!s2.empty())    {        num++;        s2.pop();    }    printf("%d\n", num);    return 0;}
0 0
原创粉丝点击