区间k大数查询

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

给定一个序列，每次询问序列中第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)，序列中的数<=106。

这道题是典型的区间查询问题，可以使用排序和查询算法来解决。

先看代码

#include 
#include 
#include 
using namespace std;

bool cmp(int a, int b) {
    return a > b;
}

int main() {
    int n, m;
    int l, r, k;
    
    while(cin >> n) {
        vector arr(n);
        for (int i = 0; i < n; i++)
            cin >> arr[i];
        cin >> m;
        while (m--) {
            cin >> l >> r >> k;
            vector brr;
            for (int j = l - 1; j < r; j++)
                brr.push_back(arr[j]);
            sort(brr.begin(), brr.end(), cmp);
            cout << brr[k - 1] << endl;
        }
    }
    return 0;
}

这道题的数据范围不是很大，不过为了更好地适应题目给定的数据规模，我建议使用动态数组（例如vector）来存储序列和临时子序列，这样可以更灵活地适应不同的数据规模。

for (int j = l - 1; j < r; j++)
 brr.push_back(arr[j]);

读入所需区间的值；

sort(brr.begin(), brr.end(), cmp);
cout << brr[k - 1] << endl;

对读入区间的数进行降序排序，sort函数默认是升序排序，但是我加了cmp函数，使得sort函数进行降序排序。因为i是从0开始的，所以输出brr[k-1]。