打卡信奥刷题（2571）用C++实现信奥 P2237 [USACO14FEB] Auto-complete S

原创于 2025-12-21 14:08:23 发布 · 483 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言 #算法 #数据库 #青少年编程

C++ 专栏收录该内容

2596 篇文章

订阅专栏

P2237 [USACO14FEB] Auto-complete S

题目描述

有 $w$ 个由小写字符构成的字典和 $n$ 个询问。每个询问由一个字符串 $s$ 和一个整数 $k$ 构成，求在字典序排序下字典内由 $s$ 为前缀的第 $k$ 字符串在输入字典的位置。若不存在，则输出 $- 1$

输入格式

第 $1$ 行是两个整数 $w$ 和 $n$

第 $2$ 行到第 $w$ + $1$ 行为字典中的第 $i$ - $1$ 个，每行由字符串构成（ $i$ 指输入的第 $i$ 行）

第 $w$ + $2$ 行到第 $w$ + $n$ + $1$ 行由一个整数 $k$ 和一个字符串 $s$ 构成

输出格式

第 $1$ 行到第 $n$ 行是对于每个询问的结果，由一个整数构成

输入输出样例 #1

输入 #1

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

输出 #1

3
1
-1

说明/提示

对于 $100\%$ 的数据， $\le 30000$ ， $1\le n \le 3000$ ，字典内每个字符串的长度均小于等于 $1000$ ，且字典的单词总长不超过 $10 ^ 6$ 。

样例解释：

对于第 $1$ 个询问，含义为在字典中找到以 a为前缀且按字典序排序后第 $4$ 个字符串，而字典中以 a为前缀且按字典序排序后为 $\{$ aa，aaa，aab，ab，abc，ac $\}$ ，第 $4$ 个是 ab，其在输入中为第 $3$ 个，故输出为 $3$

同理，对于第 $2$ 个和第 $3$ 个询问是在字典中找到以 da为前缀且按字典序排序后的第 $2$ 和第 $4$ 个字符串。而以 da为前缀的字符串按字典序排序后为 $\{$ daa，dab，dadba $\}$ ，故第 $2$ 个为 dab，其在输入中为第 $1$ 个，故第 $2$ 个输出为 $1$ ，而该序列中没有第 $4$ 个，故第 $3$ 个询问无解，输出 $- 1$

来源：USACO 2014 Feburary Contest Silver

翻译：@zymooll

C++实现

#include<bits/stdc++.h>
using namespace std;
pair<string ,int> a[90030];
bool match(string x,string y){
    if (y.length()>x.length()) return 0;
    return x.substr(0,y.size())==y;
}
int main(){
    int w,n;
    cin>>w>>n;
    for (int i=0;i<w;i++){
        cin>>a[i].first;
        a[i].second=i;
    }
    sort(a,a+w);
    while (n--){
        string pre;
        int k;
        cin>>k>>pre;
        int pos=k-1+lower_bound(a,a+w,make_pair(pre,0))-a;
        if (pos>=w || !match(a[pos].first,pre)) {
            cout<<"-1\n";
            continue;
        }
        cout<<a[pos].second+1<<'\n';
    }
    return 0;
}