AcWing 算法基础课 Trie

Trie树可以用来存储前缀字符串/数组。

可以用数组进行模拟

son[N][26] 记录节点的son

cnt[N] 记录以当前节点为最后字符的字符串出现的次数

idx当前用到的节点

 

例题 143. 最大异或对

#include<iostream>

using namespace std;

const int N=100010;

int son[N*31][2];
int idx=0;

void Insert(int a)
{
int cur=0;
for(int i=30;i>=0;i--)
{
int digit=(a>>i)&1;
if(!son[cur][digit]) son[cur][digit]=++idx;
cur=son[cur][digit];

}
}
int FindMaxXor(int a)
{
int cur=0;
int res=0;
for(int i=30;i>=0;i--)
{
res<<=1;
int digit_a=(a>>i)&1;
if(son[cur][digit_a^1]) res+=1,cur=son[cur][digit_a^1];
else cur=son[cur][digit_a];

}

return res;
}
int main()
{
int n;
cin>>n;
int res=0;
while (n -- )
{
int a;
cin>>a;
Insert(a);
res=max(res,FindMaxXor(a));
//cout<<a<<res<<endl;

}
cout<<res;
return 0;
}