Tire字符串统计

#include<iostream>
using namespace std;
const int N = 1e6 + 10;
char str[N];
int e[N][26], flag[N], idx, n; //e[N][26]用于表示树形结构，26个英文字母最多只有26个分叉
void insert(char str[]) //插入单词
{
	int p = 0; //记录头节点的位置
	for (int i = 0; str[i]; i++)
	{
		int k = str[i] - 'a'; //k表示应该插入第几个分叉
		if (!e[p][k]) //如果该结点为0(null)，并将其插入
		{
			e[p][k] = ++idx; 
		}
		p = e[p][k]; //之前子结点变成了头结点
	}
	flag[p]++; //插入成功后，遍在该单词最后一个字母的结点处+1记录
}
int query(char str[])
{
	int p = 0;
	for (int i = 0; str[i]; i++)
	{
		int k = str[i] - 'a';
		if (!e[p][k]) //如果该结点为0(null),表示没有查询到
		{
			return 0;
		}
		p = e[p][k]; //之前子结点变成了头结点
	}
	return flag[p]; //如果查询成功，返回该单词个数
}
int main()
{
	cin >> n;
	char key;
	for (int i = 0; i < n; i++)
	{
		cin >> key >> str;
		if (key == 'I')
		{
			insert(str);
		}
		else if(key=='Q')
		{
			cout << query(str) << endl;
		}
	}
	return 0;
}

刚学算法的小白的学习日记--Tire字符串统计