P2237 [USACO14FEB]自动完成Auto-complete

Posted 2021-01-01 wsmrxc

先排个序, 对于每一个询问二分前缀位置, 加上k即可.
莫名其妙WA了一次居然是因为

ios::sync_with_stdio(false);

对(puts();)有影响...

#include <cstdio>
#include <cstring>
#include <cassert>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<string, int> P;
const int MAXN = 3e4 + 10;

int W, N;
P s[MAXN];

int main()
{
    ios::sync_with_stdio(false);
    cin>>W>>N;
    for(int i = 1; i <= W; i++) cin>>s[i].first, s[i].second = i;
    sort(s + 1, s + W + 1);

    for(int i = 1; i <= N; i++){
        int k; string str;
        cin>>k>>str;
        P *pit = lower_bound(s + 1, s + W + 1, P(str, 0)) + k - 1;
        if(pit - s <= W && pit->first.substr(0, str.size()) == str)
            cout<<pit->second<<endl;
        else cout<<-1<<endl;
    }

    return 0;
}