HihoCoder - 1014 -- Trie树【字典树】

字典树

Input

输入的第一行为一个正整数n，表示词典的大小，其后n行，每一行一个单词（不保证是英文单词，也有可能是火星文单词哦），单词由不超过10个的小写英文字母组成，可能存在相同的单词，此时应将其视作不同的单词。接下来的一行为一个正整数m，表示小Hi询问的次数，其后m行，每一行一个字符串，该字符串由不超过10个的小写英文字母组成，表示小Hi的一个询问。
在100%的数据中n, m<=100000，词典的字母表大小<=26.

Output

对于小Hi的每一个询问，输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

Sample Input

5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab

Sample Output

1
0
3
0
0

思路

先将输入的单词插入到字典树中，然后统计每个结点编号遍历次数(从根节点到其他结点形成唯一的字符串)。在查询时，直接返回尾结点出现次数即可。
静态实现编号数组的大小 一般为 字符串个数 * 字符串长度

AC代码

/**
	时间复杂度为O(len*min(n, m))的算法。 
**/
/**
	时间复杂度为O(len*min(n, m))的算法。 
**/
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn = 1e6 + 5;
int tree[maxn][26];//i号结点有没有j字符的边。如果有存的是这个端点的编号 
int num[maxn];//表示出现过没有，并且出现过几次 
int k = 1;//从一号结点开始编号 
void insert(char *w, int len){
	int u = 0;
	for (int i = 0; i < len; ++i){
		int v = w[i] - 'a';
		if (!tree[u][v]){
			tree[u][v] = k++;
		}
		u = tree[u][v];
		num[u]++;
	}
}
int query(char *w, int len){
	int u = 0;
	for (int i = 0; i < len; ++i){
		int v = w[i] - 'a';
		if (!tree[u][v]){
			return 0;
		}
		u = tree[u][v];
	}
	return num[u];
}
void solve(){
	int n, m;
	char str[15];
	scanf("%d", &n);
	while (n--){
		scanf("%s", &str);
		insert(str, strlen(str));
	}
	scanf("%d", &m);
	while (m--){
		scanf("%s", &str);
		printf("%d\n", query(str, strlen(str)));
	}
}
int main(){
	solve();
	return 0;
}