区间k大数查询

问题描述

给定一个序列，每次询问序列中第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

#include<iostream>using namespace std;void function(int* collect, int length, int index, int num){for (int i = length - 2; i >= index; i--){collect[i + 1] = collect[i];}collect[index] = num;}int main(void){int n;cin >> n;int* arr = new int[n + 1];for (int i = 1; i <= n; i++){cin >> arr[i];}int m;cin >> m;int l, r, k;int* collect;while (m--){cin >> l >> r >> k;collect = new int[k];for (int i = 0; i < k; i++){collect[i] = arr[l + i];}for (int i = k - 1; i > 0; i--){for (int j = 0; j <= i - 1; j++){if (collect[j] < collect[j + 1]){collect[j] += collect[j + 1];collect[j + 1] = collect[j] - collect[j + 1];collect[j] = collect[j] - collect[j + 1];}}}for (int i = l + k; i <= r; i++){if (arr[i] > collect[k - 1]){if (k >= 2){for (int index = k - 2; index >= 0; index--){if (collect[index] > arr[i]){function(collect, k, index + 1, arr[i]);break;}else{if (index == 0){function(collect, k, 0, arr[i]);}}}}else{collect[k - 1] = arr[i];}}else{//do nothing}}cout << collect[k - 1] << endl;delete collect;}return 0;}