Cogs 647. [Youdao2010] 有道搜索框(Trie树)

1. [Youdao2010] 有道搜索框
   ★☆ 输入文件：youdao.in 输出文件：youdao.out 简单对比
   时间限制：1 s 内存限制：128 MB
   【问题描述】
   在有道搜索框中，当输入一个或者多个字符时，搜索框会出现一定数量的提示，如下图所示：
   现在给你 N 个单词和一些查询，请输出提示结果，为了简这个问题，只需要输出以查询词为前缀的并且按字典序排列的最前面的 8 个单词，如果符合要求的单词一个也没有请只输出当前查询词。
   【输入文件】
   第一行是一个正整数 N ，表示词表中有 N 个单词。
   接下来有 N 行，每行都有一个单词，注意词表中的单词可能有重复，请忽略掉重复单词。所有的单词都由小写字母组成。
   接下来的一行有一个正整数 Q ，表示接下来有 Q 个查询。
   接下来 Q 行，每行有一个单词，表示一个查询词，所有的查询词也都是由小写字母组成，并且所有的单词以及查询的长度都不超过 20 ，且都不为空
   其中： N<=10000,Q<=10000
   【输出文件】
   对于每个查询，输出一行，按顺序输出该查询词的提示结果，用空格隔开。
   【样例输入】
   youdao.in
   10
   a
   ab
   hello
   that
   those
   dict
   youdao
   world
   your
   dictionary
   6
   bob
   d
   dict
   dicti
   yo
   z
   【样例输出】
   youdao.out
   bob
   dict dictionary
   dict dictionary
   dictionary
   youdao your
   z

/*Trie.不过加了一个print过程.建树记一下单词结尾. 构造一个串输出即可.*/#include<iostream>#include<cstdio>#include<cstring>#define MAXN 100001using namespace std;int n,m,tot;char ss[MAXN],ch[MAXN];struct data{int next[27],tot;bool b;}tree[MAXN*2];void add(char s[],int l){    int now=0;    for(int i=0;i<l;i++)    {        int x=s[i]-96;        if(tree[now].next[x]) now=tree[now].next[x];        else tot++,tree[now].next[x]=tot,now=tot;    }    tree[now].b=true;}void dfs(int now,int &sum,int len){    if(sum>=8) return ;    if(tree[now].b) sum++,printf("%s ",ch);    for(int i=1;i<=26;i++)    {        if(tree[now].next[i])         {            ch[len]=i+96;            dfs(tree[now].next[i],sum,len+1);            ch[len]=0;        }    }}void query(char s[],int l){    int now=0,sum=0,x;    for(int i=0;i<l;i++)    {        x=s[i]-96;        if(!tree[now].next[x]){cout<<s<<endl;return ;}        now=tree[now].next[x];    }    memset(ch,0,sizeof ch);    for(int i=0;i<l;i++) ch[i]=s[i];    dfs(now,sum,l);    printf("\n");}int main(){    freopen("youdao.in","r",stdin);    freopen("youdao.out","w",stdout);    scanf("%d",&n);    for(int i=1;i<=n;i++)    {        cin>>ss+1;        int l=strlen(ss+1);        add(ss+1,l);    }    scanf("%d",&m);    for(int i=1;i<=m;i++)    {        cin>>ss+1;        int l=strlen(ss+1);        query(ss+1,l);    }    return 0;}