算法训练 区间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

数据规模与约定

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

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

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

#include<iostream>
#include<string>
using namespace std;

void quick(long *arr, long i, long j,long k){
	long left = i;
	long right = j;
	long key = arr[left];
	while (left < right){
		while ((left < right) && (arr[right] <= key))right--;
		arr[left] = arr[right];
		while ((left < right) && (arr[left] >= key))left++;
		arr[right] = arr[left];
	}
	arr[left] = key;
	if (left == k - 1)return;
	if ((left>k-1)&&(i < left - 1)){
		quick(arr, i, left - 1,k);
	}
	if ((left<k-1)&&(left + 1 < j)){
		quick(arr, left + 1, j,k);
	}
	
}

int main()
{
	long n;
	cin >> n;
	long *arr = new long[n + 1];
	for (long i = 0; i < n; i++){
		cin>>arr[i + 1];
	}
	long m;
	cin >> m;
	long* result = new long[m];
	long l, r, k;
	for (long i = 0; i < m; i++){
		cin >> l >> r >> k;
		long*  temporary = new long[r-l+1];
		for (long j = 0; j < r - l + 1; j++){
			temporary[j] = arr[l + j];
		}
		quick(temporary,0,r-l,k);
		result[i] = temporary[k - 1];
	}
	for (long i = 0; i < m; i++){
		cout << result[i] << endl;
	}
	return 0;
}