01字典树 + 贪心 The XOR Largest Pair LibreOJ - 10050

Posted 2021-05-25 晁棠

题目描述

在给定的 NN 个整数 A1,A2,…,ANA1,A2,…,AN 中选出两个进行异或运算，得到的结果最大是多少？

输入格式

第一行一个整数 NN。

第二行 NN 个整数 AiAi。

输出格式

一个整数表示答案。

样例

Input	Output
5 2 9 5 7 0	14

数据范围与提示

对于 100%100% 的数据，1≤N≤10^5,0≤Ai<2^31 1≤N≤10^5,0≤Ai<2^31。

 

此题用01字典树去存，每次新添加一个数，先让新的数去和Trie树异或一次贪心得到最大值，再添加进Trie树里。

 

代码：

#include <iostream>
#include <stdio.h>
#include <cstdio>
#include <algorithm>
#include <stdlib.h>
#include <cstdlib>
#include <math.h>
#include <cmath>
#include <string>
#include <string.h>
#include <cstring>
#include <vector>
#include <queue>
#include <list>
#include <stack>

using namespace std;
int T;
int N;
void ready()
{
    ios::sync_with_stdio(false),cin.tie(0);
	T=1; 
}

struct Trie_tree{
	int to[3];
	Trie_tree()
	{
		to[0]=to[1]=to[2]=0;
	}
}t[4000006];

int Ti;

long long check(long long k)
{
	int r=0;
	long long ret=0;
	for(int i=30;i>=0;i--)
	{
		long long ti=(k>>i)&1;
		if(t[r].to[ti^1]){
			r=t[r].to[ti^1];
			ret+=(1<<i);
		}
		else
		  r=t[r].to[ti];
	}
	return ret;
}

void build(long long k)
{
	int r=0;
	for(int i=30;i>=0;i--)
	{
		long long ti=(k>>i)&1;
		if(!t[r].to[ti])
		  t[r].to[ti]=++Ti;
		r=t[r].to[ti];
	}
	return ;
}

void work()
{
	long long Ans=0;
	 cin>>N;
	 for(int i=1;i<=N;i++)
	 {
	 	long long a;
	 	cin>>a;
	 	Ans=max(Ans,check(a));
	 	build(a);
	 }
	 cout<<Ans;
}

int main()
{
    ready();
    while(T--)
      work();
     return 0;
}