AcWing 每日一题 3705. 子集mex值

原题链接

数据范围
1≤T≤100,
1≤n≤100,
0≤ai≤100
输入样例：
4
6
0 2 1 5 0 1
3
0 1 2
4
0 2 0 1
6
1 2 3 4 5 6
输出样例：
5
3
4
0
样例解释
在第一个样例中，A={0,1,2},B={0,1,5}

在第二个样例中，A={0,1,2},B=∅

在第三个样例中，A={0,1,2},B={0}

在第四个样例中，A={1,3,5},B={2,4,6}

---

要想使两个子集的mex值之和最大，先尽可能的使A的mex大，然后如果A中有的元素都放在B中（相同的数放一起也不会影响mex值）

遍历两遍数组就行了

AC代码：

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int maxn=5003;
int a[maxn];
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;++i)
			scanf("%d",&a[i]);
		sort(a+1,a+n+1);
		int t=0;
		int ans=0;
		for(int i=1;i<=n;++i)
			if(a[i]==t)t++,a[i]=-1;
		ans+=t;
		t=0;
		for(int i=1;i<=n;++i)
			if(a[i]==t)t++,a[i]=-1;
		ans+=t;
		printf("%d\n",ans);
	}
	return 0;
}