输入
4
3
1 2 1
4
0 2 2 0
4
3 2 0 2
1
0
输出
Yes
Yes
No
No
mex(),原序列中没有出现的最大的数,0~n-1,这里设z=mex()
题目要求通过更改一段区间的数是否能将原来的z 更改为z+1
#include <iostream>
#include <vector>
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>
#include <queue>
#include <set>
#include <stack>
#include <algorithm>
#define x first
#define y second
#define pb emplace_back
#define fu(i,a,b) for(int i=a;i<=b; ++ i)
#define fd(i,a,b) for(int i=a;i>=b; -- i)
#define endl '\n'
#define ms(x,y) memset(x,y,sizeof x)
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef vector<vector<LL>> VVL;
typedef vector<vector<int>> VVI;
typedef vector<LL> VL;
typedef vector<bool> VB;
typedef vector<int> VI;
typedef vector<string> VS;
typedef pair<int,int> PII;
typedef vector<PII> VPII;
typedef pair<PII,int> PIII;
typedef pair<double,double> PDD;
typedef pair<double,int> PDI;
typedef pair<char,int> PCI;
typedef pair<string,int> PSI;
typedef pair<int,string> PIS;
typedef pair<int,char> PIC;
typedef pair<LL,LL> PLL;
typedef __int128 i128;
typedef unsigned long long ULL;
const int N =40 + 10 ,M = N *100 ,INF = 0x3f3f3f3f,P = 131;
const double eps = 1e-8,DNF = 1e18;
const int mod = 1e9 + 7 ,base= 20010;
const LL LNF=(LL) INF * INF;
int n;
inline void solve()
{
cin >> n ;
VI a(n);
fu(i,0,n-1) cin >> a[i];
auto mex = [](VI w) // 找到最大没有出现的数
{
sort(w.begin(),w.end());
int mx =0;
for(int x : w ) if(x == mx ) ++ mx;
return mx;
};
int z = mex(a);
if(z == n ) // 0~n-1 每个数都有
{
cout << "NO\n"; return;
}
// z == 0~ n-1
int l= find(a.begin(),a.end(),z + 1 ) - a.begin(),r= a.rend() - find(a.rbegin(),a.rend(),z + 1 ) - 1 ;
if(l == n ) // z+ 1 这个数在 原序列中没有出现过
// 把任意一个大于 z的数 改成z,mex() ==z + 1
{
cout << "YES\n";return;
}
// 将两端都是 z+1 的一段区间 修改为 z
while(l <= r ) a[l ++ ] = z;
// 再次查看是否将 mex()的值 更改为了 z + 1
cout << (mex(a)== z + 1 ? "YES\n" : "NO\n");
}
signed main()
{
// freopen("1.txt","r",stdin);
// #define int long long
// init(N-1);
ios
// cout << fixed<<setprecision(2);
int t=1;
cin>>t;
int now = 1;
while(t -- )
{
// cout<<"Case ";
// cout<<"Case #";
// cout<<"Scenario #";
// cout<< now ++ <<": ";
// cout<< now ++ <<": \n";
solve();
}
return 0;
}