题目描述
奶牛贝里斯最近有了一部新手机,于是他经常发短信。但是他经常打错单词,因为手机屏幕太小而他的爪子太大了((⊙o⊙))。农夫约翰决定帮助贝里斯来开发一个app应用,使得可以从一个不完整的单词猜想整个单词。
App应用是由W个单词组成的,每个单词都是由’a’…’z’组成的,这些单词总的长度不超过1000000。现在,总共有N个不完整的单词,每个单词的度不超过1000。对于第i个不完整的单词S_i,app应用要计算出在单词库中,按字典序排列的第K_i个前缀是S_i的单词。注意,自己也是自己的前缀。
输入
第一行是两个正整数W和N。
接下来W行,每行一个字典库里的单词。
接下里N行,每行一个K_i和其对应的不完整的单词S_i。
输出
输出包括N行,对于第i行,输出在字典库中按字典序排列的满足前缀是S_i的第K_i个单词在原字典库中的位置。如果没有足够的单词,就输出-1。
样例输入
10 3
dab
ba
ab
daa
aa
aaa
aab
abc
ac
dadba
4 a
2 da
4 da
样例输出
3
1
-1
数据范围限制
【数据规模】
字典库中单词的总长度不超过
1000000
1000000
1000000,
1
<
=
N
<
=
1000
1<=N<=1000
1<=N<=1000,W<=30000,每个不完整的单词的长度不超过
1000
1000
1000。
提示
【样例说明】
前缀是a的单词有aa,aaa,aab,ab,abc,ac,第4个是ab,ab在原字典库中是第3个。前缀是da的单词有daa,dab,dadba,第2个是dab,在原字典库中是第1个。没有第4个前缀是da的单词,所以输出-1。
解题思路
先给字典库内的单词排序,要记录原本在字典库的位置
对于每个查询的单词s,都二分一次,找到第一个出现前缀是s的单词位置
然后暴力往后枚举,找到第k个前缀是s的单词
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
using namespace std;
struct DT{
int i;
string s;
}a[30100];
int n,m,x,ans,l,r;
string s;
bool cmp(const DT&k,const DT&l){
return (k.s<l.s);
}
int main(){
freopen("auto.in","r",stdin);
freopen("auto.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
cin>>a[i].s;
a[i].i=i;//记录位置
}
sort(a+1,a+1+n,cmp);//排序
for(int i=1;i<=m;i++){
scanf("%d",&x);
cin>>s;
l=1,r=n;
while(l<r){//二分
int mid=(l+r)/2,t=0;
for(int j=0;j<s.size();j++)
if(a[mid].s[j]!=s[j]){
t=(int)(s[j])-(int)(a[mid].s[j]);
break;
}
if(t<=0)r=mid;
else l=mid+1;
}
ans=0;int j=r;
while(j<=n&&ans<x){//暴力枚举
for(int k=0;k<s.size();k++)
if(a[j].s[k]!=s[k]){
ans--;
j=n;break;
}
ans++;
j++;
}
--j;
if(ans==x)printf("%d\n",a[j].i);
else printf("-1\n");
}
}