[USACO08DEC]秘密消息Secret Message

来看一下这道题~

题目链接

题目描述

Bessie is leading the cows in an attempt to escape! To do this, the cows are sending secret binary messages to each other.

Ever the clever counterspy, Farmer John has intercepted the first b_i (1 <= b_i <= 10,000) bits of each of M (1 <= M <= 50,000) of these secret binary messages.

He has compiled a list of N (1 <= N <= 50,000) partial codewords that he thinks the cows are using. Sadly, he only knows the first c_j (1 <= c_j <= 10,000) bits of codeword j.

For each codeword j, he wants to know how many of the intercepted messages match that codeword (i.e., for codeword j, how many times does a message and the codeword have the same    initial bits). Your job is to compute this number.

The total number of bits in the input (i.e., the sum of the b_i and the c_j) will not exceed 500,000.

Memory Limit: 32MB

POINTS: 270

贝茜正在领导奶牛们逃跑．为了联络，奶牛们互相发送秘密信息．

信息是二进制的，共有M(1≤M≤50000)条．反间谍能力很强的约翰已经部分拦截了这些信息，知道了第i条二进制信息的前bi(1<=bi≤10000)位．他同时知道，奶牛使用N(1≤N≤50000)条密码．但是，他仅仅了解第J条密码的前cj(1≤cj≤10000)位．

对于每条密码J，他想知道有多少截得的信息能够和它匹配．也就是说，有多少信息和这条密码有着相同的前缀．当然，这个前缀长度必须等于密码和那条信息长度的较小者．

在输入文件中，位的总数（即∑Bi+∑Ci）不会超过500000.

输入输出格式

第1行输入N和M，之后N行描述秘密信息，之后M行描述密码．每行先输入一个整数表示信息或密码的长度，之后输入这个信息或密码．所有数字之间都用空格隔开．

明文、密码前缀的比较？这道题暴力都不那么好打...
不过明文和密码任意一位上只有0，1两种情况，我们就可以考虑用一个01Trie来维护明文，这样的话，这颗Trie上的操作也变简单了：我们只需开一个size记录一下某个根节点全部的儿子(不管是不是直接儿子)的个数就行。
不过有一个坑点，就是明文可能有重复，于是我们把bool型的标记end转成int型，就可以记录重复的次数了。
具体实现看代码吧：

#include<bits/stdc++.h>
using namespace std;

const int maxn=1000010;
int Cnt,fl,ans,num[maxn];

struct node{
	int nex[2],size,end;
	node(){
		size=0;
		nex[0]=nex[1]=0;
		end=false;
	}
}a[maxn];

int read(){
	int Value=0,Base=1;char Ch=getchar();
	for(;!isdigit(Ch);Ch=getchar())if(Ch=='-')Base=-1;
	for(;isdigit(Ch);Ch=getchar())Value=Value*10+(Ch^'0');
	return Value*Base;
}

void Build(int x,int k){
	int now=num[k];
	if(!a[x].nex[now])
		a[x].nex[now]=++Cnt;
	a[a[x].nex[now]].size++;
	if(k==fl){
		a[a[x].nex[now]].end++;
		return ;
	}
	Build(a[x].nex[now],k+1);
}

int Judge(int x,int k){
	int now=num[k];
	if(k==fl)
		return a[a[x].nex[now]].size;
	else if(a[x].nex[now] && a[a[x].nex[now]].end)
		return a[a[x].nex[now]].end+Judge(a[x].nex[now],k+1);
	if(!a[x].nex[now])return a[a[x].nex[now]].size;
	return Judge(a[x].nex[now],k+1);
}

int main( ){
	int m,n,j,k,i;
	n=read();m=read();
	for(i=1;i<=n;i++){
		fl=read();
		for(j=1;j<=fl;j++){
			num[j]=read();
		}
		Build(0,1);
	}
	for(i=1;i<=m;i++){
		fl=read();
		for(j=1;j<=fl;j++){
			num[j]=read();
		}
		ans=Judge(0,1);
		printf("%d\n",ans);
	}
	return 0;
}