CF2191B MEX Reordering

思路

将数组从小到大排序，这样前面一定会出现 $a$ 个 $0$，然后在 $a$ 之后的后缀数组的 MEX 值一定是 $0$，不会与前缀数组相同，而在 $a$ 及 $a$ 之前的前缀数组的 MEX 值一定是 $1$，只要后面有至少一个 $1$ 出现就行了。
而如果没有任何一个 $1$，且存在两个及以上的 $0$，那一定有出现以某点为分界的前缀和后缀 MEX 值都为 $1$，一定不行；若全局没有出现 $0$，则所有前缀和后缀的 MEX 值皆为 $0$，也一定不行。
所以直接按照上面判断就行了。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include<bits/stdc++.h>
#define fir first
#define sec second 
#define int long long
#define pii pair<int,int>
#define fep(i,s,e) for(int i=s;i<e;i++)
#define pef(i,s,e) for(int i=s;i>e;i--)
#define rep(i,s,e) for(int i=s;i<=e;i++)
#define per(i,s,e) for(int i=s;i>=e;i--)
namespace FastIO{
	template<typename T>inline void read(T &x){
		x=0;int f=1;char c=getchar();
		for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
		for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+(c^48);x*=f;
	}
	template<typename T,typename...Args>
    inline void read(T &x,Args&...args){
        read(x);
        read(args...);
    }
	template<typename T>void print(T x){
		if(x<0)x=-x,putchar('-');
		if(x>9)print(x/10);
		putchar((x%10)^48);
	}
}
using namespace std;
using namespace FastIO;
const int N=107;
int T,n,a[N];
signed main(){
	read(T);
	while(T--){
		read(n);
		rep(i,1,n)read(a[i]);
		int cnt1=0,cnt0=0;
		rep(i,1,n){
			if(a[i]==1)cnt1++;
			if(a[i]==0)cnt0++;
		}
		if(cnt0==0)puts("No");
		else if(cnt1==0&&cnt0>=2)puts("No");
		else puts("Yes");
	}
}