51nod 1289 大鱼吃小鱼 (模拟)(栈操作)
有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 <iostream>
using namespace std;
int a[100010] = {0};
int main()
{
int abg = 0, aed = 1;//abg 存鱼开始的地方 aed存鱼结束的地方 实时更新
int n, ta, tb;
int ans = 0;
cin>>n;
for(int i = 0; i < n ; i++)
{
cin>>ta>>tb;
if(tb == 1)
{
a[aed ++] = ta;//往右游存起来
}
else//往左游去吃往右游的
{
int tf = 1;//记录有没有全吃完
for(int j = aed - 1; j > abg; j--)
{
if(a[j] < ta)
{
a[j] = 0;//吃了
aed = j;//往右的少一个位置
}
else
{
tf = 0;//往左的吃不动了 凉了
break;
}
}
if(tf)//往左全吃完了 它活了
ans ++;
}
}
for(int i = 0; i < aed + 10; i++)
{
if(a[i] != 0)//记录还剩几条没被吃
ans ++;
}
cout<<ans<<endl;
return 0;
}
方法二: 栈操作
原理与上相同,但是代码篇幅小,易于实现
#include <iostream>
#include <stack>
using namespace std;
stack <int> stk;
int main()
{
ios::sync_with_stdio(false);
int n;
int ans = 0;
cin>>n;
int big, mode;
while(n -- )
{
cin>>big>>mode;
if(mode == 0) //吃栈
{
while(!stk.empty() && big > stk.top())
{
stk.pop();
}
if(stk.empty())
{
ans ++;
}
}
else //压栈
{
stk.push(big);
}
}
cout<<stk.size() + ans<<endl;
return 0;