1590: [Usaco2008 Dec]Secret Message 秘密信息
Time Limit: 5 Sec Memory Limit: 32 MBSubmit: 311 Solved: 216
[Submit][Status][Discuss]
Description
贝茜正在领导奶牛们逃跑.为了联络,奶牛们互相发送秘密信息.
信息是二进制的,共有M(1≤M≤50000)条.反间谍能力很强的约翰已经部分拦截了这些信息,知道了第i条二进制信息的前bi(l《bi≤10000)位.他同时知道,奶牛使用N(1≤N≤50000)条密码.但是,他仅仅了解第J条密码的前cj(1≤cj≤10000)位.
对于每条密码J,他想知道有多少截得的信息能够和它匹配.也就是说,有多少信息和这条密码有着相同的前缀.当然,这个前缀长度必须等于密码和那条信息长度的较小者.
在输入文件中,位的总数(即∑Bi+∑Ci)不会超过500000.
Input
第1行输入N和M,之后N行描述秘密信息,之后M行描述密码.每行先输入一个整数表示信息或密码的长度,之后输入这个信息或密码.所有数字之间都用空格隔开.
Output
共M行,输出每条密码的匹配信息数.
Sample Input
4 5
3 0 1 0
1 1
3 1 0 0
3 1 1 0
1 0
1 1
2 0 1
5 0 1 0 0 1
2 1 1
INPUT DETAILS:
Four messages; five codewords.
The intercepted messages start with 010, 1, 100, and 110.
The possible codewords start with 0, 1, 01, 01001, and 11.
3 0 1 0
1 1
3 1 0 0
3 1 1 0
1 0
1 1
2 0 1
5 0 1 0 0 1
2 1 1
INPUT DETAILS:
Four messages; five codewords.
The intercepted messages start with 010, 1, 100, and 110.
The possible codewords start with 0, 1, 01, 01001, and 11.
Sample Output
1
3
1
1
2
3
1
1
2
Trie树,维护一下树上前缀和就可以了
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
return x*f;
}
inline void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}
const int N=500100;
int tr[N][2],s[N],danger[N],sz=1;
int b[N],n,m;
inline void insert()
{
int now=1;
for(int i=1;i<=b[0];++i)
{
if(!tr[now][b[i]])tr[now][b[i]]=++sz;
now=tr[now][b[i]];
}
danger[now]++;
}
int sum[N];
void initial(int x)
{
sum[x]=danger[x];
if(tr[x][0])
{
initial(tr[x][0]);
sum[x]+=sum[tr[x][0]];
}
if(tr[x][1])
{
initial(tr[x][1]);
sum[x]+=sum[tr[x][1]];
}
}
inline int solve()
{
int now=1,ans=0;
for(int i=1;i<=b[0];++i)
{
now=tr[now][b[i]];
ans+=danger[now];
}
return sum[now]-danger[now]+ans;
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;++i)
{
b[0]=read();
for(int j=1;j<=b[0];++j)b[j]=read();
insert();
}
initial(1);
for(int i=1;i<=m;++i)
{
b[0]=read();
for(int j=1;j<=b[0];++j)b[j]=read();
print(solve());puts("");
}
return 0;
}
/*
4 5
3 0 1 0
1 1
3 1 0 0
3 1 1 0
1 0
1 1
2 0 1
5 0 1 0 0 1
2 1 1
1
3
1
1
2
*/