【洛谷P2922】Secret Message G【Trie树】

$link$

分析：

前缀相同字符串个数 就可以用$trie$树了
要找前缀相同个数 就在建树的时候 用$sum[]$ 看该节点有多少字符串经过
最终$ans$为 当前$ans-$当前点的$end$值$+sum$
也就是 加上前缀相同且长度比它长的字符串个数
后$：$别用结构体（

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue> 
#include<bitset>
//#pragma GCC optimize(2)
#define reg register
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int N=5e5+3;
int n,m,k,tot=1,sum[N],son[N][3],end[N];
bool a[N];
void add(bool p[])
{
	int now=1;
	for(int i=1;i<=k;i++)
	{
		int qwq=p[i];
		if(son[now][qwq]==-1) son[now][qwq]=++tot;
		now=son[now][qwq];
		sum[now]++;  //经过该节点字符串个数
	}
	end[now]++;
}
int find(bool p[])
{
	int now=1,ans=0;
	for(int i=1;i<=k;i++)
	{
		int qwq=p[i];
		if(son[now][qwq]==-1) return ans;
		now=son[now][qwq];
		ans+=end[now];  //加上以当前节点为结束的个数
	}
	return ans-end[now]+sum[now];  //加上前缀相同且长度比它长的字符串个数
}
int main(){
	scanf("%d%d",&m,&n);
	memset(son,-1,sizeof(son));
	while(m--)
	{
		scanf("%d",&k);
		for(int i=1;i<=k;i++)
			scanf("%d",&a[i]);
		add(a); 
	}
	while(n--)
	{
		scanf("%d",&k);
		for(int i=1;i<=k;i++)
			scanf("%d",&a[i]);
		printf("%d\n",find(a));
	}	
		
	return 0;
}