2020牛客暑期多校训练营（第三场）A Clam and Fish

题目链接

思路：

首先遍历一遍，数只有鱼饵和什么都没有的个数，然后第二次跑的时候如果后面的只有鱼饵或者什么都没有的池塘的个数大于手中的鱼饵数（即手中的鱼饵能够被消耗完），那么就拿鱼饵，否则就钓鱼。

代码：

#include
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=2e6+7;
const double eps=1e-8;
const int mod=1e9+7;
const int inf=0x7fffffff;
const double pi=3.1415926535;
using namespace std;
signed main()
{
    IOS;
    int t;
    cin>>t;
    while(t--)
    {
        int n,sum=0,a=0,num0=0,num1=0;
        cin>>n;
        string s;
        cin>>s;
        for(int i=0;i<n;i++)
        {
            if(s[i]=='0')
            {
                num0++;
            }
            else if(s[i]=='1')
            {
                num1++;
            }
        }
        for(int i=0;i<n;i++)
        {
            if(s[i]=='0')
            {
                if(a>0)
                {
                    a--;
                    sum++;
                }
                num0--;
            }
            else if(s[i]=='1')
            {
                if(num0+num1<=a)
                {
                    sum++;
                    a--;
                }
                else
                {
                    a++;
                }
                num1--;
            }
            else
            {
                sum++;
            }
        }
        cout<<sum<<endl;
    }
    return 0;
}