题目链接: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;
}