trie字典树【模板题】

字典树是一种实现字符串快速检索的多叉树结构。每个节点都拥有很多个指针。

• 模板提https://www.acwing.com/problem/content/144/。
• 

#include <iostream>
#include <string>
using namespace std;

const int N = 1e6 + 5, M = 5e5 + 5;

int trie[M][26], tot = 0, cnt[M];//数组模拟树，
                                //cnt[i]是用来记录以i这个节点结束的字符串数量
                                //tot是用来分配节点。
char str[N];
void insert(char* str){
	int p = 0;
	for(int i = 0; str[i]; ++ i) {
		int &s = trie[p][str[i] - 'a'];
		if(!s) s = ++ tot;//如果当前节点为空，就分配一个
		p = s;
	}
	cnt[p] ++;
}

int query(char* str) {
	int p = 0, res = 0;
	for(int i = 0; str[i]; ++ i) {
		int &s = trie[p][str[i] - 'a'];
		if(!s) break;
		p = s;
		res += cnt[p];//统计前缀字符串
	}
	return res;//返回答案
}
int main() {
	int n, m;
	cin >> n >> m;
	while(n --) {
		scanf("%s", str);
		insert(str);
	}

	while(m --) {
		scanf("%s", str);
		printf("%d
", query(str));
	}
	return 0;
}