【题解】洛谷P3879 阅读理解(trie树)

新学了字典树算法，这个可以算是一个模板吧

在有n个单词的字典中 对m个单词询问是否在此字典里。对于这个问题，我们可以通过构建字典树来解决。

trie树的思路就是对于n个单词，我们开一个数组to[maxn][26]，代表编号为maxn、字母编号为x的编号（我在说啥。。。）；具体思路就是构建一棵长成这样的树并记录其编号

然后对于读入的每个单词，我们对于其末尾编号开一个b数组记录下来代表一个单词的终结。

接着我们读入要查询的m个单词，对于每个单词的每一位来说，如果它的编号不在这棵树里，就让flag为false并退出循环。否则就继续让now递归下去。如果到了最后flag是false并且最后一位的编号是true，就说明这个单词在字典中。

 

然后对于这道题目，我的做法是构建一棵字典树，不过对判断数组进行记录他是那个句子里的，就解决了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<iomanip>
using namespace std;
const int maxn=500010;
int n,m;
int tot,now;
char s[10010];
int to[maxn][26];
bool b[maxn][1010];
//int g[1010];
void insert(int x)
{
	scanf("%s",s);
	int now=0;
	for(int i=0;i<strlen(s);i++)
	{
		if(to[now][s[i]-'a']==0)
		{
			to[now][s[i]-'a']=++tot;
		}
		now=to[now][s[i]-'a'];
	}
	b[now][x]=true;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int num;
		scanf("%d",&num);
		for(int j=1;j<=num;j++)
		{
			insert(i);
		}
	}
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		string s;
		cin>>s;
		now=0;
		bool flag=false;
		for(int j=0;j<s.size();j++)
		{
			if(to[now][s[j]-'a']==0)
			{
				flag=true;
				break;
			}
			now=to[now][s[j]-'a'];
		}
		if(flag==false)
		{
			for(int j=1;j<=n;j++)
			{
				if(b[now][j]==true)
				{
					printf("%d ",j);
				}
			}
		}
		printf("\n");
	}
	return 0;
}