解题思路
这道题挺板子的了。。
把所有的单词都存入 T r i e Trie Trie,考虑在 T r i e Trie Trie上的结点上再存一个信息 l y x [ p ] . v [ l ] lyx[p].v[l] lyx[p].v[l],表示存在或不存在一个单词属于第 l l l篇文章且单词结尾的 T r i e Trie Trie结点编号为 u u u。
对于 M M M个询问,只要查询当前询问单词在 T r i e Trie Trie树上的对应结点 u u u有哪些 p p p满足 l y x [ p ] . v [ l ] = 1 lyx[p].v[l]=1 lyx[p].v[l]=1 ( 1 < = i < = n ) (1<=i<=n) (1<=i<=n)
代码
#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<bitset>
using namespace std;
char a[30];
int n,m,nn,trie[500010][30],tot=1;
struct c{
bitset<1011> v;
}lyx[500010];
void insert(int l){
int p=1,c,len;
len=strlen(a+1);
for(int i=1;i<=len;i++)
{
c=a[i]-'a'+1;
if(!trie[p][c])
trie[p][c]=++tot;
p=trie[p][c];
}
lyx[p].v[l]=1;//标记
}
void get(){
int p=1,c,len,flag=1;
len=strlen(a+1);
for(int i=1;i<=len;i++)
{
c=a[i]-'a'+1;
if(!trie[p][c])
{
flag=0;
break;
}
p=trie[p][c];
}
if(flag)
{
for(int i=1;i<=n;i++)//查询
if(lyx[p].v[i])
printf("%d ",i);
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&m);
for(int j=1;j<=m;j++)
{
scanf("%s",a+1);
insert(i);
}
}
scanf("%d",&nn);
for(int i=1;i<=nn;i++)
{
scanf("%s",a+1);
get();
printf("\n");
}
}