题目大意: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;
}