#trie#洛谷 2922 秘密消息 Secret Message

最新推荐文章于 2025-07-12 19:29:08 发布

lemondinosaur

最新推荐文章于 2025-07-12 19:29:08 发布

阅读量174

点赞数

CC 4.0 BY-SA版权

分类专栏： Trie&AC自动机文章标签：洛谷 2922 秘密消息 Secret Message

本文链接：https://blog.youkuaiyun.com/sugar_free_mint/article/details/81806170

Trie&AC自动机专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种使用Trie树进行字符串前缀匹配的方法，通过构建Trie树并利用节点上的end标记和sum计数来高效判断一组字符串是否为另一组字符串的前缀。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

有几个字符串，现给出一些字符串 $S$ 问 $S$ 是否为字符串的前缀或字符串是 $S$ 的前缀

分析

可以用end标记末尾，用sum表示有多少字符串经过该点
然后建trie就好了

代码

#include <cstdio>
int n,m,k,tot,trie[500001][2],end[500001],sum[500001];
int in(){
	int ans=0; char c=getchar();
	while (c<48||c>57) c=getchar();
	while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
	return ans;
}
void print(int ans){
	if (ans>9) print(ans/10);
	putchar(ans%10+48);
}
int main(){
	n=in(); m=in();
	while (n--){
		k=in(); int p=0,x;
		for (register int i=1;i<=k;i++){
			x=in();
			if (!trie[p][x]) trie[p][x]=++tot;
			p=trie[p][x]; sum[p]++;
		}
		end[p]++;
	}
	while (m--){
		k=in(); int p=0,x,ans=0; bool flag=0;
		for (register int i=1;i<=k;i++){
			x=in(); p=trie[p][x];
			if (!p) flag=1; if (!flag) ans+=end[p];
		}
		if(!flag) ans+=sum[p]-end[p];//未终止
		if (ans) print(ans); else putchar('0');
		putchar('\n');
	}
	return 0;
}