【HDU1251】trie树统计前缀

1.题目链接。中文题目，就不再解释了

2.分析：这个是经典的trie树的应用。emmm，字典树就是把每个单词拆开建树，最重要的就是理解trie数组就好了。trie[i][j]=k,到底是啥意思，这里有大佬很详细的讲解了。我就不画蛇添足了字典树的介绍。不懂得可以看一看，这里我们附加一个sum数组即可，sum[i]表示i节点的数据出现的次数，最后输出sum[rt]就是答案。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e6 + 10;
int trie[maxn][26];
int sum[maxn];
int tot;
#pragma warning(disable:4996)
void insert(char*str)
{
	int rt = 0;
	int len = strlen(str);
	for (int i = 0; i < len; i++)
	{
		int x = str[i] - 'a';
		if (trie[rt][x] == 0)
			trie[rt][x] = ++tot;
		sum[trie[rt][x]]++;
		rt = trie[rt][x];
	}
}
int find(char*str)
{
	int rt = 0;
	int len = strlen(str);
	for (int i = 0; i < len; i++)
	{
		int x = str[i] - 'a';
		if (trie[rt][x] == 0)
			return 0;
		rt = trie[rt][x];
   }
	return sum[rt];
}
char s[maxn];
int main()
{
	tot = 0;
	while (gets(s))
	{
		if (s[0] == '\0')
			break;
		insert(s);
	}
	while (~scanf("%s", &s))
	{
		printf("%d\n", find(s));
	}
	return 0;
}