AcWing 835. Trie字符串统计

AcWing 835. Trie字符串统计

主要功能：

• Trie树，也就是我们常说的字典树、单词查找树。
• 是一种能够高效存储和查找字符串集合的数据结构。

我们首先规定一下：

• 把根节点设为第0个节点。
• 之后新加入一个节点先 ++idx, 再加。

为了有更好的体验，我们去手写树的创建过程：

我们看最后一张图就可以明白：

• 每次去插入元素时，如果存在该节点就接着访问下一个，若不存在则创建该节点。
• 每个红框框圈的：标记从根节点到这个节点所组成的字符串出现的次数。
• 最后查询时，我们同样从根节点出发去查询到最后，看最后一个标记的出现次数即可。

代码如下：

#include<iostream>

using namespace std;
const int maxn = 1e5 + 10;

int son[maxn][26]; // 记录每个节点
int cnt[maxn]; // 记录第 idx 个字符串 出现次数
int idx; // 用来记录字符串的编号
char str[maxn];

void insert(char str[]){
	int p = 0; // 根节点出发
	for(int i = 0; str[i] != '\0'; i ++ ) {
		int u = str[i] - 'a'; // 将26字母转换为数字，方便记录
		if(!son[p][u]) son[p][u] = ++ idx; // 子节点不存在，创建一个
		p = son[p][u]; // 走到下一个节点
	}
	cnt[p] ++; // 最重要的一行，将以p结尾的字符串出现次数 ++
}

int query(char str[]){ // 查询字符串出现次数
	int p = 0;
	for(int i = 0; str[i] != '\0'; i ++ ) {
		int u = str[i] - 'a';
		if(!son[p][u]) return 0; // 该字符串不存在，走不到最后, 次数直接为0
		p = son[p][u]; // 走到下一个节点
	}
	return cnt[p]; // 返回 以 p 结尾的字符串出现的次数
}
int main(){
	int t;
	cin >> t;
	while(t -- ){
		char op[2];
		cin >> op >> str;
		
		if(op[0] == 'I') insert(str);
		else cout << query(str) << endl;
	}
	return 0;
}