自动完成 APP
传送门 来源：upc12786
题目描述
奶牛 Bessie 很喜欢用手机上网聊天，但她的蹄子太大，经常会按到好几个键造成不必要的麻烦（丢死人了，你下辈子还是不要当奶牛了）。于是 Farmer John 给她专门设计了一套「自动完成 APP」。这个 APP 能够连接到在线词典来获取词库（词库的来源是 Bessie 常用的 n 个单词），并且有着自动补全的功能。当 Bessie 想打出她的一个常用单词时，她只需输入这个单词的某个前缀，并询问词库中拥有这个前缀的字典序第 k 小的单词，APP 就会返回这是词库中的第几个单词。
Farmer John 是个大忙人，所以编写 APP 的任务自然就交给了你。
输入
第一行两个整数 n,m，表示词库的单词个数和 Bessie 的询问次数。
接下来 n 行，每行一个字符串，表示 Bessie 的常用单词，输入的第 i 个字符串在词库中编号为 i。
接下来 m 行，每行一个整数 k 和一个字符串 S，表示 Bessie 的询问，意义是询问词库中以 S 为前缀的字典序第 k 小的单词在词库中的编号。
输出
对于每次询问输出一个整数，表示以S为前缀的字典序第 k 小的单词在词库中的编号；如果不存在这样的单词，输出 −1。
样例输入
10 3
dab
ba
ab
daa
aa
aaa
aab
abc
ac
dadba
4 a
2 da
4 da
样例输出
3
1
-1
提示：
(1) 以 a 为前缀的单词有 {aa,aaa,aab,ab,abc,ac}，其中字典序第 4 小的是 ab，是词库的第 3 个单词。
(2) 以 da 为前缀的单词有 {daa,dab,dadba}，其中字典序第 2 小的是 dab，是词库的第 1 个单词；没有第 4 小的单词。
对于 20% 的数据，1≤ 词库的字母总量,询问的字母总量 ≤103。
对于 100%的数据，1≤ 词库的字母总量,询问的字母总量 ≤106，保证词库里的单词两两不同，所有字符串只包含小写字母。
思路：
如果对字典树有了解，开始能感觉到是个字典树的板子题加上dfs能解决，但TLE。
看了题解，先用字典树把所有单词存起来，再跑一次dfs给每个单词的结尾附上id（连续的），就可以直接求了，具体看解析。
思路解析：（输入单词：dab ba ab daa aa aaa）
1、建立字典树（具体怎么建字典树：浅谈Trie树（字典树）写的很好）
用sum数组记录每个点是几个单词的前缀，例如sum[4]=1。
如果单词结束还要用flag1数组表示他是第几个单词，便于最后输出，例如flag1[10]=6（因为他是第六个单词嘛）。
这两点在建字典树的时候经常用到，不难理解。
上图中红色点表示其上面字符串构成了一个单词。
2、dfs给红色点赋id（蓝色数字为赋给的id，有规律便于结题）
蓝色的数字表示赋给的id，存入数组num中，如果是红点的话还要用数组dir存入其root值（root值就是圈中心的数字）。
例如 root=3时 id=6 则：
num[3]=6 dir[6]=3;
root=1 时 id=5
只需要num[1]=5而不需要存入dir数组中，因为root不是红点。
3.1、根据输入的字符串判断是否有存在单词满足该前缀的可能性
例如输入：1 fffffff (这不是乔碧萝殿下的坦克！！)
从图中可以看出不可能构成单词，可以直接输出-1。
3.2、如果满足条件，返回root的值
例如输入：3 b
返回root的值： 4 判断sun[4]和k的大小关系，如果不满足输出-1。
4、如果上面条件不能输出-1，那就意味着肯定能找到第k小的单词。
直接输出flag1[ dir[ num[ root ] + k - 1 ] ]就是结果。我当时就看到就很懵逼，但分析完这个公式你会感觉整个解题思路超级紧密。
分析公式：flag1[ dir[ num[ root ] + k - 1 ] ]
1、num[ root ] + k - 1 表示的是一个数的 id
2、把id放进num[ ]中即 num[id]是不是就表示这个数的root值
3、再把root放进flag1[]中即flag1[root]是不是表示第几个输入的单词
类似链式前向星的head数组和edge数组的关系，就是很巧妙。
分析完公式，超级佩服这个思路，感谢：溺生。
AC代码：
#include<bits/stdc++.h>
using namespace std;
const int MAX=1e6;
int trie[MAX+5][35];
char a[MAX+5];
int n,m,k;
int top;
int flag[MAX+5],flag1[MAX+5],num[MAX+5],dir[MAX+5];
int sum[MAX+5];
void Insert(char a[],int k1) ///建字典树
{
int la=strlen(a);
int root=0;
for(int i=0;i<la;i++){
int id=a[i]-'a';
if(!trie[root][id]) trie[root][id]=++top;
sum[trie[root][id]]++;
root=trie[root][id];
}
flag[root]=1; ///标记该root点有单词结束
flag1[root]=k1; ///给root点赋输入的顺序，便于输出
}
int cnt;
void dfs(int res) ///dfs给所有的root点赋上图id值
{
if(flag[res]){
dir[++cnt]=res;
num[res]=cnt;
}else{
num[res]=cnt+1;
}
for(int i=0;i<26;i++){
if(trie[res][i]){
dfs(trie[res][i]);
}
}
}
int Find(char a[])///判断输入的前缀是不是有构成单词的可能性，如果有返回最后的root值，没有返回-1
{
int la=strlen(a);
int root=0;
for(int i=0;i<la;i++){
int id=a[i]-'a';
if(!trie[root][id]){
return -1;
}
root=trie[root][id];
}
return root;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%s",a);
Insert(a,i+1);
}
dfs(0);
for(int i=0;i<m;i++){
scanf("%d%s",&k,a);
int res=Find(a);
if(res==-1){
printf("-1
");
}else{
if(sum[res]<k){
printf("-1
");
}else{
printf("%d
",flag1[dir[num[res]+k-1]]);///输出结果
}
}
}
return 0;
}