【Ybtoj 第9章例4】阅读理解【Trie树】

---

解题思路

这道题挺板子的了。。

把所有的单词都存入$Trie$，考虑在$Trie$上的结点上再存一个信息$lyx[p].v[l]$，表示存在或不存在一个单词属于第$l$篇文章且单词结尾的$Trie$结点编号为$u$。
对于$M$个询问，只要查询当前询问单词在$Trie$树上的对应结点$u$有哪些$p$满足$lyx[p].v[l]=1$ $(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");
	}
}