51 Nod 大鱼吃小鱼
题目来源: 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<bits/stdc++.h>
using namespace std;
int a[100010];
int b[100010];
int lmax[100010];//当前位置的左边向右的鱼的最大值
int rmax[100010];//当前位置的右边向左的鱼的最大值
int main()
{
//freopen("in.txt","r",stdin);
int N;
cin>>N;
for(int i=0;i<N;i++)cin>>a[i]>>b[i];
int max_l=-1;int max_r=-1;
lmax[0]=-1;
rmax[N-1]=-1;
for(int i=0;i<N-1;i++)
{
if(b[i]==1)
{
if(a[i]>max_r)max_r=a[i];
}
else
{
if(a[i]>max_r)max_r=-1;
}
lmax[i+1]=max_r;
}
for(int i=N-1;i>=1;i--)
{
if(b[i]==0)
{
if(a[i]>max_l)max_l=a[i];
}
else
{
if(a[i]>max_l)max_l=-1;
}
rmax[i-1]=max_l;
}
int ans=0;
for(int i=0;i<N;i++)
{
if(b[i]==0)
{
if(a[i]>=lmax[i])ans++;
}
else
{
if(a[i]>=rmax[i])ans++;
}
}
cout<<ans<<endl;
return 0;
}