算法基础之数据结构——Trie字符串统计

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

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

输入格式
第一行包含整数N，表示操作数。
接下来N行，每行包含一个操作指令，指令为”I x”或”Q x”中的一种。

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

数据范围
1≤N≤2∗1e4

输入样例：
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例：
1
0
1

思想：
题目有两种操作，一种操作是向集合中插入字符串，另外一种是查询集合中字符串出现的次数。根据算法思想，我们建立一颗树，每一个字符串中的第一个字符就是根节点的子节点，然后字符串往后寻找它的子节点，如果树中没有这个节点，我们就创建一个，直到字符串末尾。这样，我们在字符串末尾的那个节点记录字符串出现的频率。

例如：
如下图：

第一个字符串为abcdef，那么就创建节点a,b,c,d,e,f，并且我们在最后字符串结尾“f”记录一下字符串出现的频率（其余的字符串如图，同理）

最后不难得出代码：

#include<iostream>

using namespace std;

const int N = 100005;
//son是节点，因为每一个节点最多出现25个字母，所以设为son[N][25]，count是记录最后字符串结尾出现的频率
int son[N][26], count[N], flag;//flag是记录当前数组的下标（因为这是用数组模拟的树）
char str[N];
//插入操作
void insert(char str[])
{
    int p = 0;
    for(int i = 0; str[i]; i ++){
        int x = str[i] - 'a';
        if(!son[p][x])  son[p][x] = ++ flag;
        
        p = son[p][x];
    }
    count[p] ++ ;
}
//查询操作
int query(char str[])
{
    int p = 0;
    for(int i = 0; str[i]; i ++){
        int x = str[i] - 'a';
        if(!son[p][x])  return 0;
        p = son[p][x];
    }   
    return count[p];
}

int main()
{
    int n;
    cin >> n;
    while( n -- ){
        char ins[2];
        cin >> ins >> str;
        
        if(ins[0] == 'I')  insert(str);
        else    cout << query(str) << endl;
    }
    return 0;
}