算法训练 区间k大数查询

时间:2022-09-09 22:43:20
问题描述

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