Trie树(字典树) 板子

Trie树用于高效插查字符串 题目中一定会有字符的范围(如26个字母等)

利用二维数组实现 核心三点son[N][26]: N代表待分配池；26代表根据当前字符选择存放位置；存放的数为下一个结点的N。

eg：维护字符串集合 询问某字符串在集合中的出现次数

#include<bits/stdc++.h>
using namespace std;

const int N=100010;

int son[N][26], cnt[N], idx;// 
char str[N];

void insert(char *str){  
	int p=0;                               从0开始
	for(int i=0;str[i];i++){
		int u=str[i]-'a';                  选择存放位置
		if(!son[p][u]) son[p][u]=++idx;    初始为0 所以之后从idx=1开始分配
		p=son[p][u];                       当前结点更新到刚分配好的结点
	}
	cnt[p]++;
}

int query(char *str){
	int p=0;
	for(int i=0;str[i];i++){
		int u=str[i]-'a';
		if(!son[p][u]) return 0;
		p=son[p][u];
	}
	return cnt[p];
}

int main(){
	int n;
	scanf("%d",&n); // getchar();
	while(n--){
		char op[2];
		scanf("%s%s",op,str);
		if(*op=='I') insert(str);
		else cout<<query(str)<<endl;
	}
} 

eg: 最大异或对

首先 需要能熟悉TRIE树插入 查找的时间复杂度 每一次插入查找都是常数复杂度

其次 TRIE树的待分配池大小的界定: 直接设置为所有字符数

核心一点： 已知某数二进制下的总位数 后续操作 左右移

#include<bits/stdc++.h>
using namespace std;

const int N=100010;
const int M=3100000;// 一共存10e5*31
int a[N];
int t[M][2];
int idx;

void insert(int a){
	int p=0;
	for(int i=30;~i;i--){
		int x=a>>i&1;   //知道总共的二进制位数 抽取二进制位！！！ 
		if(!t[p][x]) t[p][x]=++idx;
		p=t[p][x];
	}
}

int query(int a){
	int p=0;
	int res=0;
	for(int i=30;~i;i--){
		int x=a>>i&1;
		if(t[p][!x]){
			res+=1<<i;
			p=t[p][!x];
		}
		else p=t[p][x];
	}
	return res;
} 

int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
		insert(a[i]);
	}
	int res=0;
	for(int i=0;i<n;i++){
		int x=query(a[i]);
		if(x>res) res=x;
	}
	cout<<res;
}