chenchen题解：Trie字符串统计

题目描述

维护一个字符串集合，支持两种操作：

1.“I x”向集合中插入一个字符串x；
“Q x”询问一个字符串在集合中出现了多少次。
共有N个操作，输入的字符串总长度不超过 10^5 ，字符串仅包含小写英文字母。

输入格式

第一行包含整数N，表示操作数。

接下来N行，每行包含一个操作指令，指令为“I xIx”或“Q xQx”中的一种。

数据范围：

$$1<n<2\ast 10^4$$

输出格式

对于每个询问指令“Q x，都要输出一个整数作为结果，表示x在集合中出现的次数。

每个结果占一行。

Sample Input

5
I abc
Q abc
Q ab
I ab
Q ab

Sample Output 1

1
0
1

算法思想

1、插入过程 （I操作）
对于一个单词，从根开始，沿着单词的各个字母所对应的树中的节点分支向下走，直到单词遍历完，将最后的节点标记一下，表示该单词已插入Trie树。
2、查询过程 （Q操作）
同样的，从根开始按照单词的字母顺序向下遍历trie树，一旦发现某个节点没有标记或者单词遍历完成而最后的节点未标，则表示该单词不存在。若最后的节点标记了，表示该单词存在。

时间复杂度

$$O(mn\ast logn)$$

代码实现

#include <iostream>
using namespace std;
const  int N=1000010;
int son[N][30],cnt[N];
int n;
int idx;
char s[N];
void in(char s[])
{
	int p=0;
	for(int i=0;s[i];i++)
	{
		int u=s[i]-'a';
		if(!son[p][u]) son[p][u]=++idx;
		p=son[p][u];
	}
	cnt[p]++;
}
int qu(char s[])
{
	int p=0;
	for(int i=0;s[i];i++)
	{
		int u=s[i]-'a';
		if(!son[p][u]) return 0;
		p=son[p][u];
	}
	return cnt[p];
}
int main(int argc, char** argv) 
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		string op;
		cin>>op>>s;
		if(op=="I")
		{
			in(s);
		}
		else
		{
			cout<<qu(s)<<endl;
		}
	}
	return 0;
}

理解万岁！！