Codeforces 842 D Vitya and Strange Lesson 线段树

传送门：Codeforces 842D

题意：给出一个长度为n的序列和m次操作，每次操作将序列中所有数异或上给定的数x，然后求序列的mex。

mex：给定序列中未出现的最小的非负数。

思路：首先不要被题目中说的每次序列都会变化迷惑，因为异或满足结合律，因此我们只要把所有的x异或起来，就相当于对原序列和一个x求题目中的要求，假设原序列为a[i]，和x异或以后得到b[i]，我们要求mex(b[i])，就是要在a序列中找一个没出现过且二进制每一位都尽量和x相等的数（先保证最高位相等，然后，以此类推），这一点结合异或和mex的性质可以得出，由于a[i]和x都不超过3e5，因此我们可以建立一个完全二叉树来表下一位示每一个数的出现情况，这颗二叉树其实就是一个二进制树，每一层对应一个数字的二进制中的一位，左子树代表0，右子树代表1，(类似于字典树，不过是只有01两个字符)，然后维护每一个结点对应的子树中出现的数的个数，询问的时候就按照上面的红字分析，我们要找的数首先要尽量和x相等，因此我们要按照x的二进制位走，其次要没出现过，因此当我们碰到某个子树中的数全部出现时，我们就必须换另一颗子树走（换子树走的时候就会对答案产生该层对应二进制位权值的贡献）。

代码：

#include<bits/stdc++.h>
#define ll long long
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
#define inf 0x3f3f3f3f
using namespace std;
typedef pair<int,int> P;
const int N = (1 << 19) - 1;
const int MAXN = 600010;
int cnt[MAXN << 2];
int dig[20];
void update(int x, int l, int r, int rt)
{
	if(l == r)
	{
		cnt[rt] = 1;
		return ;
	}
	int mid = (l + r) >> 1;
	if(x <= mid)
	update(x, lson);
	else
	update(x, rson);
	cnt[rt] = cnt[rt << 1] + cnt[rt << 1 | 1];
}
int query(int dep, int l, int r, int rt)
{
	if(l == r) return 0;
	int mid = (l + r) >> 1;
	if(dig[dep])
	{
		if(cnt[rt << 1 | 1] == r - mid)
		return query(dep - 1, lson) + (1 << dep);
		return query(dep - 1, rson);
	}
	else
	{
		if(cnt[rt << 1] == mid - l + 1)
		return query(dep - 1, rson) + (1 << dep);
		return query(dep - 1, lson);
	}
}
int main()
{
	int n, m, t;
	cin >> n >> m;
	for(int i = 0; i < n; i++)
	scanf("%d", &t), update(t, 0, N, 1);
	while(m--)
	{
		scanf("%d", &t);
		for(int i = 0; i <= 18; i++)
		if(t & 1 << i)
		dig[i] ^= 1;
		printf("%d\n", query(18, 0, N, 1));
	}
 	return 0;
}