Trie字符串统计

Trie字符串统计

问题需求

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

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

输入格式
第一行包含整数N，表示操作数。

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

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

每个结果占一行。

数据范围
1≤N≤2∗10⁴

代码解析

#include<iostream>
using namespace std;
const int N = 100010;
int son[N][26],cnt[N],idx=0;//下标为0的点，既是根节点，又是空节点
char str[N];              //son[N][26]表示的是纵向的
void insert(char str[])
{
    int p=0;
    for(int i=0;str[i];i++)//遍历整个字符串，直到str[i]==0
    {
        int u=str[i]-'a';
        if(!son[p][u])son[p][u]=++idx;//字节点为空的时候新建
        p=son[p][u];
    }
    cnt[p]++;
}
int query(char str[])
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(!son[p][u])return 0;
        p=son[p][u];
    }
    return cnt[p];
}
int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        char op[2];
        scanf("%s%s",op,str); 
        /*1.C风格字符串只要是使用双引号（半角）的，都会以\0结束，就是说会多出一个字符。
        2.凡是数组没有指名大小的定义，全部都会以赋值号右边的初始化长度作为数组默认大小。
        字符串"abcd"长度为4，是因为字符串长度指的是字符串中有效字符的个数，不包含C风格结束符\0，
        而字符串数组a[]="abcd"长度为5，因为这个是算的数组的长度，也就是所占用的存储空间，
        当然要包含\0结束符（它也要占用存储空间）。
        a[10]="abcd"的长度为10，因为你显示指定了字符数组的长度，所占内存当然就是10，
        它的存储空间是连续的，也就是说，10个字符空间中，前面分别是a,b,c,d和结束符\0，
        后面全部都是0（被初始化了），总共有5个0，加上前面的abcd和结束符\0，一共10个，他们的存储是连续的。
        a[]={'a','b','c','d'}，没有按照双引号定义，因此不算C风格字符串，只是一个字符数组。
        而且又没有指明大小，那么它的大小等于你初始化的长度，这里就是4，sizeof也等于4，没有结束符\0了。*/
        //printf("op[0]:%c ,op[1]:%d \n",op[0],op[1]);
        if(op[0] == 'I') insert(str);
        else printf("%d\n",query(str));
    }
    return 0;
}