Contest1802 - 2019年第二阶段我要变强个人训练赛第十八场 问题 E: 自动完成 APP 排序+二分

题目链接：http://icpc.upc.edu.cn/problem.php?cid=1802&pid=4

问题 E: 自动完成 APP

时间限制: 1 Sec  内存限制: 128 MB
提交: 48  解决: 6
[提交] [状态] [命题人:admin]

题目描述

奶牛 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，保证词库里的单词两两不同，所有字符串只包含小写字母。

题解：刚开始直接字典树，搞了好长时间，超时，用指针就超内存，想了一下在保存上是没问题的，但是查询时，直接查询到叶节点，会超时的，然后可以发现，字母的数量是1e6所以直接排序就行了，然后给出前缀直接二分找到第1个，然后直接判断输出第k个即可

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
struct node {
    string s;
    int id;
}a[N];
int n, m;
bool cmp(node x, node y) {
    return x.s < y.s;
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie();
     
    cin >> n >> m;
    for(int i = 1; i <= n; i++) {
        cin >> a[i].s;
        a[i].id = i;
    }
    sort(a + 1, a + 1 + n, cmp);
    int k;
    string s;
    int l, r, mid, ans, len, flag;
    while(m--) {
        cin >> k >> s;
        len = s.size();
        l = 1, r = n;
        while(l <= r) {
            mid = (r + l ) >> 1;
            if(a[mid].s >= s) {
                ans = mid;
                r = mid - 1;
            } else l = mid + 1;
        }
        if(ans + k - 1 > n) {
            printf("-1\n");
            continue;
        }
        ans = ans + k - 1;
        flag = 1;
        for(int i = 0; i < len; i++) {
            if(s[i] == a[ans].s[i]) continue;
            flag = 0;
            break;
        }
        if(flag) printf("%d\n", a[ans].id);
        else printf("-1\n");
    }
    return 0;
}