标签：using name build 出现 insert 一个 algo char 字符串统计
AcWing 835. Trie字符串统计 https://www.acwing.com/problem/content/837/
维护一个字符串集合，支持两种操作：
共有N个操作，输入的字符串总长度不超过 105105，字符串仅包含小写英文字母。
第一行包含整数N，表示操作数。
接下来N行，每行包含一个操作指令，指令为”I x”或”Q x”中的一种。
对于每个询问指令”Q x”，都要输出一个整数作为结果，表示x在集合中出现的次数。
每个结果占一行。
1≤N≤2∗1041≤N≤2∗104
5
I abc
Q abc
Q ab
I ab
Q ab
1
0
1
#include <iostream> #include <cstdio> #include <algorithm> using namespace std; const int N = 1e5+5; int son[N][26], cnt[N], index; char str[N]; void build(char str[]) { int p = 0; for(int i = 0; str[i]; ++ i) { int u = str[i] - ‘a‘; if(!son[p][u]) son[p][u] = ++ index; 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; cin >> n; while(n --) { char op[2]; scanf("%s%s", op, str); if(*op == ‘I‘) build(str); else printf("%d\n", query(str)); } return 0; }
N = int(1e5 + 5) son = [[0 for i in range(26)] for i in range(N)] cnt = [0 for i in range(N)] index = 0 def insert(x): global index p = 0 for i in range(len(x)): t = ord(x[i]) - ord(‘a‘) if not son[p][t]: index += 1 son[p][t] = index p = son[p][t] cnt[p] += 1 def query(x): p = 0 for i in range(len(x)): t = ord(x[i]) - ord(‘a‘) if not son[p][t]: return 0 p = son[p][t] return cnt[p] n = int(input()) while n: n -= 1 t = input().split() op = t[0] s = t[1] if op == ‘I‘: insert(s) else: res = query(s) print(res)
标签：using name build 出现 insert 一个 algo char 字符串统计
原文地址：https://www.cnblogs.com/Chaosliang/p/12233073.html
