【洛谷P2922】Secret Message

题目大意：给定 N 个字符串组成的字典，有 M 个询问，每次给定一个字符串，求字典中有多少个单词为给定字符串的前缀或前缀是给定的字符串。

题解：在 trie 上维护一个 tag 表示有多少字符串以当前字符串为前缀，ed 表示从该节点到根节点之间为一个字符串。

对于给定的字符串，在 trie 中顺序查找，在匹配的过程中累加经过节点的 ed 标记，表示有多少字符串为给定字符串的前缀。若匹配成功，则答案累加 tag 标记并减去最后一个节点的 ed 标记，防止重复计算。若匹配失败，只需输出记录的 ed 之和即可。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;

int trie[maxn][2],tot=1,tag[maxn],ed[maxn];
int m,n,len,s[10010];

void insert(int *ss){
	int now=1;
	for(int i=1;i<=len;i++){
		int ch=ss[i];
		if(!trie[now][ch])trie[now][ch]=++tot;
		now=trie[now][ch],++tag[now];
	}
	++ed[now];
}
void query(int *ss){
	int now=1,ret=0;
	for(int i=1;i<=len;i++){
		int ch=ss[i];
		if(!trie[now][ch]){
			printf("%d\n",ret);
			return;
		}
		now=trie[now][ch];
		ret+=ed[now];
	}
	printf("%d\n",ret+tag[now]-ed[now]);
}

int main(){
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++){
		scanf("%d",&len);
		for(int j=1;j<=len;j++)scanf("%d",&s[j]);
		insert(s);
	}
	for(int i=1;i<=n;i++){
		scanf("%d",&len);
		for(int j=1;j<=len;j++)scanf("%d",&s[j]);
		query(s);
	}
	return 0;
}