首页 > 题解 > bzoj 1590 [Usaco2008 Dec]Secret Message 秘密信息

bzoj 1590 [Usaco2008 Dec]Secret Message 秘密信息

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.

Sample Output

1

3

1

1

2

HINT

0 matches only 010: 1 match 1 matches 1, 100, and 110: 3 matches 01 matches only 010: 1 match 01001 matches 010: 1 match 11 matches 1 and 110: 2 matches

题解

裸的trie树啊。。就是建树的时候每个点做过的时候cnt[i]++,查询的时候中间的加以这个点结尾的串的数量,结尾的时候加cnt

#include <cstdio>
#include <cstring>
#define N 500010
#define getchar() (S == T && (T = (S = BB) + fread(BB, 1, 1 << 15, stdin), S == T) ? EOF : *S++)
using namespace std;
char BB[1 << 15], *S = BB, *T = BB;
int inline read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int ch[N][2],sums[N],cnt[N],a[N],root=1,sz=1;
void add(int a[],int n)
{
    int now=root;
    for (int i=1;i<=n;i++)
    {
        if (ch[now][a[i]])
            now=ch[now][a[i]];
        else
            now=ch[now][a[i]]=++sz;
        cnt[now]++;
    }
    sums[now]++;
}
int asks(int a[],int n)
{
    int now=root,ans=0;
    for (int i=1;i<=n;now=ch[now][a[i]],i++)
        if (!ch[now][a[i]])
            return ans;
        else if (i!=n)
            ans+=sums[ch[now][a[i]]];
        else
            ans+=cnt[ch[now][a[i]]];
    return ans;
}
main()
{
    int n=read(),m=read();
    for (int i=1;i<=n;i++)
    {
        int k=read();
        for (int j=1;j<=k;j++)
            a[j]=read();
        add(a,k);
    }
    for (int i=1;i<=m;i++)
    {
        int k=read();
        for (int j=1;j<=k;j++)
            a[j]=read();
        printf("%d\n",asks(a,k));
    }
}

榜上第三233