字典树——operator_

例题：P8306 【模板】字典树

概述

想象一下字典，你查字典肯定不是从头翻到尾的吧，字典树就借鉴了这个思想。

每个节点下连接了几个儿子，而节点本身代表一个字符（根除外），那么从一个节点到根节点的路径，就是代表的一个字符串。

它有一些显然的性质，比如同前缀的字符串会处在一个子树内，可以利用这些性质干一些事（比如本题就可以 $dfs$ 求子树和，做到 $O(1)$ 询问）。

它可以做到 $O(|s|)$ 的插入与查找。

代码

#include<bits/stdc++.h>
using namespace std;
int T,n,q;
string s;
int f(char c) {
	if(c>='0'&&c<='9') return c-'0';
	if(c>='a'&&c<='z') return c-'a'+10;
	if(c>='A'&&c<='Z') return c-'A'+36;
}
int t[3000005][65],cnt;
int a[3000005];
void insert(string s) {
	int p=0,l=s.length();
	for(int i=0;i<l;++i) {
		int r=f(s[i]);
		if(!t[p][r])
			t[p][r]=++cnt;
		p=t[p][r];
		++a[p];
	}
}
int query(string s) {
	int p=0,l=s.length();
	for(int i=0;i<l;++i) {
		int r=f(s[i]);
		if(!t[p][r])
			return 0;
		p=t[p][r];
	}
	return a[p];
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>T;
	while(T--) {
		cin>>n>>q;
		for(int i=0;i<=cnt;++i) {
			for(int j=0;j<=63;++j)
				t[i][j]=0;
			a[i]=0;
		}
		cnt=0;
		while(n--) {
			cin>>s;
			insert(s);
		}
		while(q--) {
			cin>>s;
			printf("%d\n",query(s));
		}
	}
	return 0;
}

我真是算法越简单越不知道啊