BZOJ1590: [Usaco2008 Dec]Secret Message 秘密信息

题目大意：给n条01信息以及m条01密码，求对于每条密码，有多少条信息与他的最长公共前缀=min(密码长度，该条信息长度)

先对于信息建出一颗字典树，终止节点权值+1，然后每条密码的答案=路径上所有点的权值和+终止节点子树(不包括终止节点)权值总和

所以我们可以建出Trie树之后pushup一下，就能得到每个点子树的权值总和，这样就可以O（密码长度）回答每个询问了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 500010
using namespace std;
int ch[N][2],w[N],sum[N],cnt=1;
void addnew()
{
	int nn,i,j,x,y;
	scanf("%d",&nn);
	int now=1;
	for(i=1;i<=nn;i++)
	{
		scanf("%d",&x);
		if(!ch[now][x])
		{
			cnt++;
			ch[now][x]=cnt;
		}
		now=ch[now][x];
	}
	w[now]++;
}
void pup(int x)
{
	sum[x]=w[x];
	if(ch[x][0])
	{
		pup(ch[x][0]);
		sum[x]+=sum[ch[x][0]];
	}
	if(ch[x][1])
	{
		pup(ch[x][1]);
		sum[x]+=sum[ch[x][1]];
	}
}
void checkit()
{
	int nn,i,j,x,y;
	scanf("%d",&nn);
	int now=1,ans=0;
	for(i=1;i<=nn;i++)
	{
		scanf("%d",&x);
		now=ch[now][x];
		ans+=w[now];
	}
	printf("%d\n",ans-w[now]+sum[now]);
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	int i;
	for(i=1;i<=n;i++)
	addnew();
	pup(1);
	for(i=1;i<=m;i++)
	checkit();
}