Codeforces Round #779 (Div. 2)-D2 - 388535 (Hard Version)

传送门：https://codeforces.com/contest/1658/problem/D2

思路：字典树
首先答案res一定位于 a[i]^l 之中，那么可以通过遍历 a[i]^l 来求得res, 但直接暴力遍历会超时，可通过建立字典树，然后求 a[i]^l 异或字典树的最大值 Max 和最小值 Min, 由于正确答案res^a[i]时会得到 [l,r]区间，因此只要 Min=l,Max=r那么就是正确答案

Code:

#include<iostream>
using namespace std;

const int N=17;
const int MAX_N=2e5+5;
const int MAX_S=3200005;
int n,Q,T;
int Trie[MAX_S][2],num;
bool Sum[MAX_S];
int a[MAX_N];

void Insert(int x);
int QueryMin(int x);
int QueryMax(int x);
int main()
{
	cin>>T;
    int l,r;
	while(T--){
		for(int i=0;i<=num;++i)
			Trie[i][0]=Trie[i][1]=Sum[i]=0;
		num=0;
		cin>>l>>r;
        n=r-l+1;
		for(int i=0;i<n;++i)
		{
			cin>>a[i];
			Insert(a[i]);
		}
		int res=0;
        for(int i=0;i<n;++i)
        {
            res=a[i]^l;
            if(QueryMin(res)==l&&QueryMax(res)==r){
                break;
            }
        }
        cout<<res<<endl;
	}

	return 0;
}

void Insert(int x)
{
	int u=0,t;
	for(int i=N;i>=0;--i)
	{
		t=(x>>i)&1;
		if(!Trie[u][t])	Trie[u][t]=++num;
		u=Trie[u][t];
		Sum[u]=1;
	}
}

int QueryMax(int x)
{
	int u=0,t,p,res=0;
	for(int i=N;i>=0;--i)
	{
		t=(x>>i)&1;
		p=Trie[u][!t];
		if(p&&Sum[p]){
			res=res|(1<<i);
			u=p;
		}else{
			u=Trie[u][t];
		}
	}
	return res;
}

int QueryMin(int x)
{
	int u=0,t,p,res=0;
	for(int i=N;i>=0;--i)
	{
		t=(x>>i)&1;
		p=Trie[u][t];
		if(p&&Sum[p]){
			u=p;
		}else{
			res=res|(1<<i);
			u=Trie[u][!t];
		}
	}
	return res;
}