Tire树学习笔记及练习题

Trie树

主要参考：宝藏

1、模板题
建立一棵trie树即可。
trie树两个基本操作是插入和查找。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int nex[300001][27];
int exist[300001];

void solve(){
	int n, m, cnt = 0;
	cin >> n;
	for(int i = 1;i <= n;i ++){
		string s;
		cin >> s;
		int len = s.length(), p = 0;
		for(int k = 0;k < len;k ++){
			int c = s[k] - 'a';
			if(!nex[p][c]) nex[p][c] = ++ cnt;
			p = nex[p][c];
		}
		exist[p] = 1;
	}
	cin >> m;
	for(int i = 1;i <= m;i ++){
		string s;
		cin >> s;
		int len = s.length(), p = 0;
		for(int k = 0;k < len;k ++){
			int c = s[k] - 'a';
			if(!nex[p][c]) break;
			p = nex[p][c];
		}
		if(exist[p] == 1){
			exist[p] ++;
			cout << "OK" << endl;
		}
		else if(exist[p] == 2){
			cout << "REPEAT" << endl;
		}
		else cout << "WRONG" << endl;
	}
	return;
}

int main(){
	solve();
	return 0;
}

2、01trie树
题目
给你一颗树，求两个结点的最大异或值。

根据异或的性质，与同一个数异或两次，数不变。因此可以任取一个点作为根，记T(u,v)为u到v的异或值，那么T(u,v) = T(root,u) ^ T(root,v),因此求出每一个点到根的异或值，然后插入01trie树中。
在trie树中，数是从高位到低位在树中存储。寻找最大的异或值即可。对于a的最大异或值，要求从根出发时，尽量走不同的路径。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int MAXN = 2000001;
vector<pair<int, int> >G[MAXN];
int w[MAXN];
int nex[MAXN][2];
int vis[MAXN];

void dfs(int now){
	for(int i = 0;i < G[now].size();i ++){
		int v = G[now][i].first;
		w[v] = w[now] ^ G[now][i].second;
		dfs(v);
	}
	return;
}

int find(int x){
	int p = 0;
	for(int i = 31;i >= 0;i --){
		int v = (x >> i) & 1;
		if(v == 1){
			if(nex[p][0]) p = nex[p][0];
			else p = nex[p][1];
		}
		else if(v == 0){
			if(nex[p][1]) p = nex[p][1];
			else p = nex[p][0];
		}
	}
	return vis[p];
}

void solve(){
	int n;
	cin >> n;
	for(int i = 1;i < n;i ++){
		int f,t,c;
		cin >> f >> t >> c;
		G[f].push_back({t, c});
	}
	dfs(1);
	int cnt = 0;
	for(int i = 1;i <= n;i ++){
		int p = 0;
		for(int k = 31;k >= 0;k --){
			int t = (w[i] >> k) & 1;
			if(!nex[p][t]) nex[p][t] = ++ cnt;
			p = nex[p][t];
		}
		vis[p] = w[i];
	}
	int ans = -1;
	for(int i = 1;i <= n;i ++)
		ans = max(ans, w[i] ^ find(w[i]));
	cout << ans << endl;
	return;
}

int main(){
	solve();
	return 0;
}