Acwing 835. Trie字符串统计

Acwing 835. Trie字符串统计

题目描述

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

1. I x 向集合中插入一个字符串 x；
2. Q x 询问一个字符串在集合中出现了多少次。

共有 N
个操作，输入的字符串总长度不超过 105，字符串仅包含小写英文字母。

输入格式

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

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

输出格式

对于每个询问指令 Q x，都要输出一个整数作为结果，表示 x

在集合中出现的次数。

每个结果占一行。

数据范围

1≤N≤2∗104

思路

用树来存储字符串，存储的方法就是用二维的数组 f[i][j] , i 代表了第几个字母，比如的第一个存入acwing 那么对于 f[0][‘a’-‘a’] = 等于下一个字母的索引，即 1 ；然后f[1][‘c’-‘a’] = 下一个字母的索引，f[2] 对应的就是 字母 w -> f[2][‘w’-‘a’]。 结尾用另外一个数组来记录，cnt ,如果这个字母是结尾，即f[i][j] 是结尾 那么cnt[f[i][j]] 加一，记录出现的次数。

代码

#include<iostream>
#include<algorithm>
using namespace std;
#include<string>
const int N = 100010 ;

int tire[N][26],cnt[N] ;

int idx ;
char op[100010],str[N];
void insert(char* str )
{
    int j = 0;
    for(int i = 0 ;str[i] ; ++ i)
    {
        int u = str[i] - 'a' ;
         if(!tire[j][u])
          tire[j][u] =   ++ idx ;
          j = tire[j][u];
    }
    cnt[j] ++ ;
}

int is(char * ch)
{
    int j = 0;
    for(int i = 0 ;ch[i] ; ++ i)
    {
        int u = ch[i] - 'a';
        if(tire[j][u]) j = tire[j][u];
        else return 0 ;
    }
    
    return cnt[j];
}
int main()
{
    int n;
    scanf("%d", &n);
    while (n -- )
    {
        char op[2];
        scanf("%s%s", op, str);
        if (*op == 'I') insert(str);
        else printf("%d\n", is(str));
    }

    return 0;
}