Codeforces Round #602 Div2 D1. Optimal Subsequences (Easy Version)

题意:给你一个数组a,询问m次,每次返回长度为k的和最大的子序列(要求字典序最小)的pos位置上的数字.

题解:和最大的子序列很简单,排个序就行,但是题目要求字典序最小,那我们在刚开始的时候先记录每个数的位置再排序,然后选出k个最大的数后在对位置从小到大排个序就行了(这题有个坑,第一次排序的时候记得把相等的数按位置小的排在前面).

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
 
int n,m;
int k,pos;
PII a[N];
vector<PII> tmp;
 
bool cmp1(PII a,PII b){
    if(a.fi==b.fi) return a.se<b.se;
    return a.fi>b.fi;
}
 
bool cmp2(PII a,PII b){
    return a.se<b.se;
}
 
int main() {
    ios::sync_with_stdio(false);
    cin>>n;
     for(int i=0;i<n;++i){
         cin>>a[i].fi;
        a[i].se=i;
     }
     sort(a,a+n,cmp1);
    cin>>m;
     while(m--){
         cin>>k>>pos;
         tmp.clear();
         for(int i=0;i<k;++i) tmp.pb(a[i]);
        sort(tmp.begin(),tmp.end(),cmp2);
        printf("%d\n",tmp[pos-1].fi);
     }
 
    return 0;
}

总结:学会使用pair!!