前缀树板子

合集 - 算法模板(30)

1.01穷举模板（dfs）01-052.单调栈板子01-103.拓补排序板子（邻接表实现）01-114.堆结构与堆排序01-145.LCS(递归/记忆化/dp）01-186.树状数组板子（单点增加+范围查询）01-207.位运算内容01-248.线段树板子01-25

9.前缀树板子01-26

10.树状数组板子02-1611.数论分块02-1912.ST表模板（RMQ查询）02-2213.扩展欧几里得算法模板02-2314.快速幂模板02-2315.同余式/乘法逆元/费马小引理02-2316.矩阵乘法/矩阵快速幂模板02-2417.分解质因数02-2418.埃氏筛/欧拉筛模板02-2419.欧拉函数/欧拉定理/扩展欧拉定理02-2520.拓补排序板子（卡恩算法/02-2521.最短路Djikstra算法板子02-2522.LIS板子（二分优化）02-2523.LCS板子02-2524.SPFA板子02-2625.prim板子02-2726.并查集板子01-0427.Kruskal板子02-2728.线段树板子03-0429.字符串哈希板子03-0930.KMP板子03-09

收起

build方法（注意将节点编号cnt初始化为1）

int cnt;
const int maxn=1e6+5;
int trie[maxn][26];
int pass[maxn];
int last[maxn];
void build(){
	cnt=1;
}

insert方法（向字典树中插入一个单词）

void insert(string word){
	int cur=1;
	pass[cur]++;
	for(int i=0,path;i<word.size();i++){
		path=word[i]-'a';
		if(trie[cur][path]==0){
			trie[cur][path]=++cnt;
		}
		cur=trie[cur][path];
		pass[cur]++;
	}
	last[cur]++;
}

search方法（查询单词出现频率）

int search(string word){
	int cur=1;
	for(int i=0,path;i<word.size();i++){
		path=word[i]-'a';
		if(trie[cur][path]==0){
			return 0;
		}
		cur=trie[cur][path];
	}
	return last[cur];
}

prefix方法（查询以pre作前缀的单词数量）

int prefix(string pre){
	int cur=1;
	for(int i=0,path;i<pre.size();i++){
		path=pre[i]-'a';
		if(trie[cur][path]==0){
			return 0;
		}
		cur=trie[cur][path];
	}
	return pass[cur];
}

erase方法（在trie中删除单词word）

void erase(string word){
	if(search(word)>0){
		int cur=1;
		for(int i=0,path;i<word.size();i++){
		path=word[i]-'a';
		if(--pass[trie[cur][path]]==0){
			trie[cur][path]=0;
			return;
		}
		cur=trie[cur][path];		
		}
		last[cur]--;
	}
}