ICPC（自动完成APP）（ STL应用、重载结构体、lower_bound() 在结构体中的使用 ）

问题 E: 自动完成 APP

时间限制: 1 Sec  内存限制: 256 MB
提交: 52  解决: 5
[提交] [状态] [命题人: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，保证词库里的单词两两不同，所有字符串只包含小写字母。

题解

输入N个字符串后，排个序，同时记录他在未排序之前的编号。所以这需要结构体。然后就是对结构体排序。之后需要在最快时间内找到前缀为s的第一个字符串，需要使用二分中 的lower_bound（） 所以我们需要运用重载运算符，重新定义查找规则。

值得注意的是lower_bound（）中第三个参数也必须是一个结构体，所以你需要将前缀字符串s封装成结构体才可以查询。

#include <bits/stdc++.h>
using namespace std;
struct node
{
	int i;
	string s;
	bool operator<(const node m)const{
        return s<m.s;
    }
}hah[1000006];
char ch[1000006];
bool cmp(struct node x,struct node y)
{
	return x.s<y.s;
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++)
    {
    	scanf("%s",ch);
    	hah[i].s=ch;
    	hah[i].i=i;
	}
	sort(hah,hah+n,cmp);
	while(m--)
	{
		string c;
		int k;
		struct node kk;
		scanf("%d %s",&k,ch);
		c=ch;
		kk.s=c;
		int tt=lower_bound(hah,hah+n,kk)-hah;
		if(hah[tt+k-1].s.find(c)!=0)printf("-1\n");
		else printf("%d\n",hah[tt+k-1].i+1);
	}
    return 0;
}