试题 算法训练 区间k大数查询
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
给定一个序列,每次询问序列中第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。
方法:使用一个辅助数组,每次只排序[l-1, r)的部分,然后返回倒数第k个数即可
#include <iostream>
#include <algorithm>
void slove()
{
int n; // 序列长度
int m; // 询问个数
std::cin >> n;
int * arr = new int[n]; // 给定的序列
int * temp = new int[n]; // 辅助序列
for (int i = 0; i < n; ++i)
{
std::cin >> arr[i];
}
std::cin >> m;
int L, r, k;
for (int i = 0; i < m; ++i)
{
std::cin >> L >> r >> k;
// 复制
for (int j = L - 1; j < r; ++j)
{
temp[j] = arr[j];
}
// 排序
std::sort(temp + L - 1, temp + r);
std::cout << temp[r - k] << '\n';
}
delete arr;
arr = nullptr;
}
int main()
{
slove();
return 0;
}