在给定的 N 个整数 A1,A2……AN 中选出两个进行 xor(异或)运算,得到的结果最大是多少?
输入格式
第一行输入一个整数 N。
第二行输入 N 个整数 A1~AN。
输出格式
输出一个整数表示答案。
数据范围
1≤N≤105,
0≤Ai<231
输入样例:
3
1 2 3
输出样例:
3
如上图,假设输入的是123,那么对应二进制异或结果为3,2,1,其中最大的结果为3.
我们先尝试用暴力算法求解,然后思考利用数据结构进行优化。
这样做的时间复杂度是
O
(
n
2
)
O(n^2)
O(n2),显然不满足要求。因此我们尝试通过研究二进制数异或的特点来优化这个算法。
思路:对于一个数
A
i
A_i
Ai,我们希望在
A
1
−
A
n
A_1-A_n
A1−An中找到与它异或结果最大的数。
如上图,通过观察,我们发现最适合的数就是从最高位开始往下遍历,每一位都与
A
i
A_i
Ai不同的数字。因此我们将整个集合拿来遍历,每走一步,我们都删除当前位数字符与
A
i
A_i
Ai相同的子集,保留其余剩下的数字,直到集合无法再被划分。这样的操作正让我们想到了树的自顶向下遍历。
如上图,首先我们根据数据构建一个串树,这个串树每条边仅由0或1构成,然后针对每个具体数(在这里我们拿2作为例子),根据上面的思想对树进行一个向下的查找。2的二进制是010(注意所所有数字要统一位数),因此从根节点出发,我们渴望走的是“1”,这条边,然后接下来如图中红箭头所示,最终得到的结果是5,因此对于数字2,在现有数据中与2异或结果最大的数字是5,计算结果为7。
其余计算如下
.
比较特殊的是6(110),它在第二位数字只有1这一条边可以走。最终所有计算结果最大值为7,因此我们返回7.
#include<bits/stdc++.h>
using namespace std;
#define fic(i,a,b) for(int i=a;i>=b;i--)
#define fir(i,a,b) for(int i=a;i<=b;i++)
const int N=100100;
int trie[N*32][2],tot=1,a[N],n,ans;
void insert(int x)
{
int p=1;
fic(k,30,0)
{
int ch=x>>k&1;
if (trie[p][ch]==0)
trie[p][ch]=(++tot);
p=trie[p][ch];
}
}
int Search(int x)
{
int p=1,ans=0;
fic(k,30,0)
{
int ch=x>>k&1;//取出第k位
if (trie[p][ch^1])//异或走路
{
p=trie[p][ch^1];
ans|=(1<<k);
}
else
p=trie[p][ch];
}
return ans;
}
int main()
{
cin>>n;
fir(i,1,n)
{
scanf("%d",&a[i]);
insert(a[i]);
ans=max(ans,Search(a[i]));//取最大值
}
cout<<ans<<endl;
}
作者:秦淮岸灯火阑珊
链接:https://www.acwing.com/solution/content/923/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。