tyvj1228 有道搜索框

P1228 有道搜索框

时间: 10000ms / 空间: 131072KiB / Java类名: Main

描述

在有道搜索框中，当输入一个或者多个字符时，搜索框会出现一定数量的提示，如下图所示：

现在给你N个单词和一些查询，请输出提示结果，为了简这个问题，只需要输出以查询词为前缀的并且按字典序排列的最前面的8个单词，如果符合要求的单词一个也没有请只输出当前查询词。 

输入格式

第一行是一个正整数N，表示词表中有N个单词。
接下来有N行，每行都有一个单词，注意词表中的单词可能有重复，请忽略掉重复单词。所有的单词都由小写字母组成。
接下来的一行有一个正整数Q，表示接下来有Q个查询。
接下来Q行，每行有一个单词，表示一个查询词，所有的查询词也都是由小写字母组成，并且所有的单词以及查询的长度都不超过20，且都不为空
其中：N<=10000,Q<=10000 

输出格式

对于每个查询，输出一行，按顺序输出该查询词的提示结果，用空格隔开。 

测试样例1

输入

10 
a 
ab 
hello 
that 
those 
dict 
youdao 
world 
your 
dictionary 
6 
bob 
d 
dict 
dicti 
yo 
z

输出

bob 
dict dictionary 
dict dictionary 
dictionary 
youdao your 
z

分析

        trie树裸题啊，

        对于读入，直接建字典树，对于询问，直接进行遍历，直到输出满8个字符串即可

        技巧：对于字符串的输出，可以建一个栈，每次走过一条边就把上面的字母进栈，每次走到有标记的点就把整个栈输出，每次回溯就弹一次栈。

        关于Trie树的知识，我以后可能会把它和AC自动机一起总结

代码

//tyvj1228 有道搜索框 trie树
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 200005
using namespace std;
int N, Q, cnt;
char s[250], stack[maxn], top;
int tot=1, e[maxn][27], end[maxn], len;
void insert(int x)
{
	int p=1, i;
	while(x<len)
	{
		if(e[p][s[x]-'a'])p=e[p][s[x++]-'a'];
		else p=e[p][s[x++]-'a']=++tot;
	}
	end[p]=true;
}
void dfs(int p, int x)
{
	int i;
	if(x==len && end[p] && cnt)
	{
		cnt--;
		for(i=1;i<=top;i++)putchar(stack[i]);
		putchar(' ');
	}
	if(x<len)
	{
		if(cnt && e[p][s[x]-'a'])
		{
			stack[++top]=s[x];
			dfs(e[p][s[x]-'a'],x+1);
			top--;
		}
	}
	else
	{
		for(i=0;i<26 && cnt;i++)
		{
			if(e[p][i])
			{
				stack[++top]=i+'a';
				dfs(e[p][i],x);
				top--;
			}
		}
	}
}
int main()
{
	int i;
	scanf("%d",&N);
	for(i=1;i<=N;i++)
	{
		scanf("%s",s);
		len=strlen(s);
		insert(0);
	}
	scanf("%d",&Q);
	for(i=1;i<=Q;i++)
	{
		scanf("%s",s);
		len=strlen(s);
		cnt=8;
		dfs(1,0);
		if(cnt==8)printf("%s",s);
		putchar('\n');
	}
	return 0;
}