题目意思比较简单。
解题思路:对于2和3的情况只需捕鱼。
对于0情况,如果有饲料则捕鱼。
对于1情况,我们开一个数组记录每个位置后面所剩的0,1情况个数,如果现有饲料个数已经超过此位置记录的个数,就捕鱼,否则就增加饲料。
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
typedef long long ll;
ll a[2000007];
int main()
{
ll t;
cin>>t;
while(t--)
{
ll fish=0,bait=0,num=0;
ll n;
cin>>n;
string s;
cin>>s;
if(s[0]=='0'||s[0]=='1')
a[0]=1;
for(int i=1;i<s.length();i++)
{
a[i]=a[i-1];
if(s[i]=='0'||s[i]=='1')
{
a[i]+=1;
}
}
for(int i=0;i<s.length();i++)
{
if(s[i]=='0')
{
if(bait>0)
{
fish++;
bait--;
}
}
else if(s[i]=='1')
{
if(bait>=a[n-1]-a[i]+1&&bait!=0)
{
fish++;
bait--;
}
else
{
bait++;
}
}
else if(s[i]=='2')
{
fish++;
}
else if(s[i]=='3')
{
fish++;
}
}
cout<<fish<<endl;
}
return 0;
}