A. LuoTianyi and the Show

原题链接

题解

• 如果第一个出场的是 \(-2\) 那么，接下来只能用 \(-2\) 和大于零的数
• 如果第一个出场的是 \(-1\) 那么，接下来只能用 \(-1\) 和大于零的数
• 如果第一个出场的是大于零的数，那么接下来的 \(-1\) 可以补全其左边的空，\(-2\) 可以补全其右边的空

code

#include<bits/stdc++.h>
using namespace std;
int vis[100005]={0};
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        int n,m;
        cin>>n>>m;

        vector<int> ans;
        int cnt=0,l=0,r=0;
        for(int i=1;i<=n;i++)
        {
            int x;
            cin>>x;
            if(x>0&&!vis[x]&&(vis[x]=1)) ans.push_back(x);
            else if(x==-1) l++;
            else if(x==-2) r++;
        }
        sort(ans.begin(),ans.end());

        for(int i=0;i<ans.size();i++)
        {
            cnt=max(cnt,1+min(ans[i]-1,i+l)+min(m-ans[i],(int)ans.size()-i-1+r));//如果以它先开始
        }

        cnt=max(cnt,max(min(m,(int)ans.size()+l),min(m,(int)ans.size()+r)));
        cout<<cnt<<"\n";
        memset(vis,0,sizeof vis);
    }
    return 0;
}