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