正题
评测记录:https://www.luogu.org/recordnew/lists?uid=52918&pid=P2922
题目大意
给n个01串,用m个01串匹配,如果n是m的前缀或者m是n的前缀那么就算匹配成功,求每个串有多少个匹配成功。
解题思路
我们可以用Trie树,分为0和1分支,然后在路上统计下面有多少个匹配。
code
#include<cstdio>
using namespace std;
int n,m,nn,end[500001],l[500001][2],x,tot,sum[500001];
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++){scanf("%d",&nn);int now=0;for(int i=1;i<=nn;i++){scanf("%d",&x);if(!l[now][x]) l[now][x]=++tot;//新建节点now=l[now][x];//下一个点sum[now]++;//路上统计}end[now]++;//结束统计}for(int i=1;i<=m;i++){scanf("%d",&nn);int now=0,ans=0,j;bool flag=false;for(j=1;j<=nn;j++){scanf("%d",&x);if(!l[now][x]||flag) {flag=true;continue;}//不再匹配now=l[now][x];//下一个点ans+=end[now];//统计答案}if(!flag)ans+=sum[now]-end[now];//特判printf("%d\n",ans);}
}