区间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;
}