【网易有道10年编程赛 资格赛第一场】 第二题 有道搜索框

描述

 

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


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

输入

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

输出

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

样例输入

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

 

#include <iostream> #include <string> #define MAX_N 1000000 using namespace std; struct dic { bool isword; int pos[27]; }dics[MAX_N]; int N, Q, index = 1; void buildTree(int p, const string &str, int curindex) { int len = str.length(); if(p >= len) { dics[curindex].isword = true; return; } int t = int(str[p] - 'a') + 1; if(dics[curindex].pos[t] == 0) dics[curindex].pos[t] = ++index; buildTree(p + 1, str, dics[curindex].pos[t]); } void travel(int &tempCount, int pos, int curindex, const string &str, string res) { int i, len = str.length(); if(pos <= len - 1) { int t = int(str[pos] - 'a') + 1; if(dics[curindex].pos[t] == 0) return; travel(tempCount, pos + 1, dics[curindex].pos[t], str, res); } else { if(tempCount == 0) return; if(dics[curindex].isword) { cout<<res<<" "; tempCount--; } for(i = 1; i <= 26; i++) { if(dics[curindex].pos[i] != 0) { travel(tempCount, pos, dics[curindex].pos[i], str, res + char(i - 1 + 'a')); } } } } int main() { int i, j; string str; cin>>N; for(i = 0; i < N; i++) { cin>>str; buildTree(0, str, 0); } cin>>Q; for(i = 0; i < Q; i++) { cin>>str; int countv = 8; travel(countv, 0, 0, str, str); if(countv == 8) cout<<str<<endl; else cout<<endl; } return 0; }