acwing-835.Trie字符串统计

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

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

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

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

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

每个结果占一行。

数据范围

1≤N≤2∗10000

输入样例：

5
I abc
Q abc
Q ab
I ab
Q ab

输出样例：

1
0
1

分析

int son[N][26], cnt[N], idx;
就是一个就是一个邻接表

son[N][26]就是分配的N个节点给trie使用，同时每个节点有26个状态，每次只有一个状态是成立的，而表头就是son[0][26]每个状态对应一个开头
而insert操作就是一个尾插法，有头节点的链表尾插就要遍历到最后一个元素。这个for循环就是在做这个操作

void insert(char str[])
{
    int p = 0; //从根结点开始遍历
    for (int i = 0; str[i]; i ++ )
    {
        int u =str[i] - 'a';
        if (!son[p][u]) son[p][u] = ++ idx; //没有该子结点就创建一个
        p = son[p][u]; //走到p的子结点
    }

    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];
}
就是一次邻接表的遍历过程，如果路径不通就返回0

代码

#include<bits/stdc++.h>
#include<algorithm>
using namespace std;
const int N = 100010;
int son[N][26],cnt[N];
int n, p, idx;

//插入 
void insert(char str[]){
   p =	0;//目前所在的节点 
   //遍历字符串里面的字符 
   for(int i = 0; str[i]; i++){
   	//26个字母，0~25 
   	   int u = str[i] - 'a';
   	   //以abc为例 
   	   //P等于0，如果0的子节点，a不存在，idx++，idx 就相当于存储空间的下标 
   	   if(!son[p][u]) son[p][u] = ++idx;
   	   p = son[p][u];//告诉p你下一步往哪里走 
   }
   cnt[p] ++;//这个字符串遍历完，记住他插入的次数 
}
//查询 
int  query(char str[]){
	p = 0;
	for(int i = 0; str[i]; i++){
		//u的作用，u代表下一个字母，是唯一的，这也使得p是唯一的 
		int u = str[i] - 'a'; 
		//如果这个字母的下一个字母不存在,直接返回 
		if(!son[p][u]) return 0;
		p = son[p][u]; 
	}
	return cnt[p];//返回重复次数 
}
int main(){
	cin >> n;
	char c;
	char x[N];
	while(n--){
		cin >> c;
		if(c == 'I'){
			cin >> x;
			insert(x);
		}else if(c == 'Q'){
			cin >> x;
			printf("%d \n",query(x));
		}
	}
	return 0;
}