算法训练 区间k大数查询（题解）

资源限制

　　时间限制：1.0s   内存限制：256.0MB

 

问题描述

　　给定一个序列，每次询问序列中第l个数到第r个数中第K大的数是哪个。

 

输入格式

　　第一行包含一个数n，表示序列长度。

　　第二行包含n个正整数，表示给定的序列。

　　第三个包含一个正整数m，表示询问个数。

　　接下来m行，每行三个数l,r,K，表示询问序列从左往右第l个数到第r个数中，从大往小第K大的数是哪个。序列元素从1开始标号。

 

输出格式

　　总共输出m行，每行一个数，表示询问的答案。

 

样例输入

5
1 2 3 4 5
2
1 5 2
2 3 2

 

样例输出

4
2

 

数据规模与约定

　　对于30%的数据，n,m<=100；

　　对于100%的数据，n,m<=1000；

　　保证k<=(r-l+1)，序列中的数<=10^6。

 

本题可以借助STL的algorithm和vector将数据进行排序，然后查找并输出。

代码如下：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
bool cmp(int a, int b)
{
    return a > b;
}
vector<int> a;
vector<int> b;
int n, m, l, r, k;
int main()
{
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        int t;
        cin >> t;
        a.push_back(t);
    }
    cin >> m;
    while (m--)
    {
        int t;
        cin >> l >> r >> k;
        for (int i = l - 1; i < r; i++)
        {
            b.push_back(a[i]);
        }
        sort(b.begin(), b.begin() + (r - l + 1), cmp);
        cout << b[k - 1] << endl;
        b.clear();
    }
    return 0;
}