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

题目链接

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

题解：把信息建一棵Trie，分别标记路径结点cov和单词结点dang
查询的时候把密码在Trie上跑一遍，未到结尾对应将答案累加dang（密码比信息长），到结尾将答案累加cov（密码比信息短）

我的收获：Trie基本应用

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=500015;

int n,m;

int s[N];

struct Trie{

#define idx s[i]

int tot;
int dang[N],cover[N];
int c[N][13];

void insert(int len){
    int x=0;
    for(int i=0;i<len;i++){
        if(!c[x][idx]) c[x][idx]=++tot;
        x=c[x][idx];cover[x]++;
    }
    dang[x]++;
}

void query(int len){
    int x=0,sum=0;
    for(int i=0;i<len;i++){
        if(!c[x][idx]) break;
        x=c[x][idx];sum+=(i!=len-1)?dang[x]:cover[x];
    }
    printf("%d\n",sum);
}

}T;

void init()
{
    scanf("%d%d",&n,&m);
    for(int o,i=1;i<=n;i++){
        scanf("%d",&o);
        for(int j=0;j<o;j++) scanf("%d",&s[j]);
        T.insert(o);
    }
    for(int o,i=1;i<=m;i++){
        scanf("%d",&o);
        for(int j=0;j<o;j++) scanf("%d",&s[j]);
        T.query(o);
    }
}

int main()
{
    init();
    return 0;
}