51nod 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

用栈来存储b[i]=1的鱼(向右移动的,要去战斗的),当遇到b[i]=0的鱼(向左移动的,来挑战的),双方血拼,用ans记录牺牲的鱼。
最终剩下的鱼=n-ans;

 1 #include <iostream>
 2 #include <stack>
 3 using namespace std;
 4 int a[100005];
 5 int b[100005];
 6 stack<int> sta;
 7 int main()
 8 {
 9     int n;
10     int ans=0;
11     cin>>n;
12     for(int i=0;i<n;i++)
13     {
14         cin>>a[i];
15         cin>>b[i];
16     }
17     for(int i=0;i<n;i++)
18     {
19         if(b[i]==0&&sta.empty())
20             continue;
21         if(b[i]==1)
22         {
23             sta.push(a[i]);
24             continue;
25         }
26         while(!sta.empty())
27         {
28             int tmp=sta.top();
29             sta.pop();
30             if(tmp>a[i])
31             {
32                 ans++;
33                 sta.push(tmp);
34                 break;
35             }
36             ans++;
37         }
38     }
39     cout<<n-ans<<endl;
40     return 0;
41 }
View Code
 
原文地址:https://www.cnblogs.com/onlyli/p/7258206.html