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
#include
#include
#define MAXN 100001
using 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;iint 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;i96;
        if(!tree[now].next[x]){cout<return ;}
        now=tree[now].next[x];
    }
    memset(ch,0,sizeof ch);
    for(int i=0;iprintf("\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;
}

你可能感兴趣的:(树结构)