自动补全（春季每日一题 58）

奶牛贝茜买了一个新手机，并十分喜欢用它发短信。

但是它的大蹄子在手机的小屏幕上打字时遇到了麻烦，它总是把单词拼错。

农夫约翰同意通过编写一个自动补全程序来帮助它，当它输入部分单词时，该应用程序可以给出补全建议。

自动补全程序可以访问一个包含 $W$ 个单词的词典，每个单词由 $a..z$ 范围内的小写字母组成，其中所有单词中的字母总数最多为 $1000000$。

现在给定一个包含 $N$ 个部分单词的列表，每个部分单词最多包含 $1000$

对于每个部分单词 $i$，还会提供一个整数 $K_i$，自动补全程序需要找到词典中以部分单词 i 作为前缀的所有单词中，按字典序排序排在第 $K_i$

也就是说，对所有部分单词 i 的有效补全按字典序进行排序，输出此顺序下的第 $K_i$

输入格式
第一行包含两个整数 $W$ 和 $N$。

接下来 $W$

接下来 $N$ 行，每行首先包含一个整数 $K_i$，然后包含一个部分单词 $i$。

输出格式
共 $N$ 行，第 $i$ 行输出部分单词 $i$ 的第 $K_i$ 个有效补全在词典中出现的位置（一个 $1∼W$

如果第 $K_i$ 个有效补全不存在，则输出 ​​​−1​​。

数据范围
$1≤W≤30000,$
$1≤N≤10000$
$1≤K_i≤W$

输入样例：

10 3
dab
ba
ab
daa
aa
aaa
aab
abc
ac
dadba
4 a
2 da
4 da

输出样例：

3
1
-1

样例解释
​​​a​​​ 的自动补全为 ​​aa,aaa,aab,ab,abc,ac​​​，第四个为 ​​ab​​，在词典中排第 3 个。

​​da​​​ 的自动补全为 ​​daa,dab,dadba​​​，第二个为 ​​dab​​，在词典中排第 1 个。

​​da​​ 不存在第四个自动补全。

---

排序 + 二分 + hash

#include<iostream>
#include<algorithm>
#include<unordered_map>

using namespace std;

const int N = 30010;

int w, n;
string str[N];
unordered_map<string, int> mp;

int get(string s){
    
    int l = 0, r = w + 1;
    while(l < r){
        int mid = l + r >> 1;
        if(str[mid] >= s) r = mid;
        else l = mid + 1;
    }
    return l;
}

int main(){
    
    cin >> w >> n;
    
    for(int i = 1; i <= w; i++)
        cin >> str[i], mp[str[i]] = i;
    
    sort(str + 1, str + w + 1);
    
    while(n--){
        
        int idx;
        string s;
        cin >> idx >> s;
        
        int p = get(s);
        if(p == 0 || p == w + 1 || p + idx - 1 > w) puts("-1");
        else{
            string s2 = str[p + idx - 1];
            if(s == s2.substr(0, s.size())){
                cout << mp[str[p + idx - 1]] << endl;
            }else puts("-1");
        }
    }
    
    return 0;
}