P2922 [USACO08DEC]Secret Message G(Trie)

最新推荐文章于 2023-12-05 22:22:37 发布

酷酷的Herio

最新推荐文章于 2023-12-05 22:22:37 发布

阅读量315

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Trie

本文链接：https://blog.csdn.net/weixin_45750972/article/details/118523054

Trie 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了如何利用Trie树解决USACO竞赛中的字符串最大前缀匹配问题。通过构建Trie树并维护节点信息，可以高效地进行模式串与文本串的匹配，实现O(n)时间复杂度的查询。代码示例展示了具体的实现过程。

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

P2922 [USACO08DEC]Secret Message G(Trie)

Trie 可用用于统计字符串的个数和前缀和匹配等问题。

题意：给定 $m$ 个文本串， $n$ 个模式串。

求每个模式串与多少个文本串的最大前缀相同。

对 $m$ 个文本串构造 $T r i e$ ，然后每次 $O (n)$ 查找即可。

建 $T r i e$ 时，维护两个变量，一个是经过该结点的次数 $s$ ，一个是以该结点为结尾的字符串个数 $e d$ 。

然后计算贡献的时候，跑一遍字符串即可，每次加上ed，若没遍历完则此时的sum就是答案，否则还要加上a[p].s-a[p].ed，a[p].s包括了a[p].ed,而a[p].ed之前被统计过，所以只需加上真前缀为该模式串的文本串个数。

code

// Problem: P2922 [USACO08DEC]Secret Message G
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P2922
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// Date: 2021-07-06 15:48:40
// --------by Herio--------

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=5e4+5,M=5e5+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define fi first
#define se second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define IOS ios::sync_with_stdio(false),cin.tie(0) 
void Print(int *a,int n){
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\n",a[n]); 
}
struct node{
	int nt[2],s,ed;
}a[M];
int b[N],len,cnt;
void ins(){
	int p=0;
	for(int j=1;j<=len;j++){
		if(!a[p].nt[b[j]]) a[p].nt[b[j]]=++cnt;
		p=a[p].nt[b[j]];
		a[p].s++;
	}
	a[p].ed++;
}
ll que(){
	ll ans=0;
	int p=0;
	bool ok=true;
	for(int i=1;i<=len;i++){
		if(!a[p].nt[b[i]]){
			ok=0;break;
		} 
		p=a[p].nt[b[i]];
		ans+=a[p].ed;
	}
	return ok?ans+a[p].s-a[p].ed:ans;
}
int main(){
	int n,m;scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++){
		scanf("%d",&len);
		for(int j=1;j<=len;j++) scanf("%d",&b[j]);
		ins();
	}
	for(int i=1;i<=n;i++){
		scanf("%d",&len);
		for(int j=1;j<=len;j++) scanf("%d",&b[j]);
		printf("%lld\n",que());
	}
	return 0;
}