蓝桥杯练习-算法训练-区间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。
解题思路
暴力: 将给定区间的数储存到优先队列,或者其他能排序的数据结构,再一次取出k次,最后一个输出。
AC代码
#include<iostream>
#include<algorithm>
#include<string.h>
#include<cmath>
#include<queue>
using namespace std;
int main() {
std::ios::sync_with_stdio(false);
int n,m;
int a[1050];
while (cin >> n) {
for (int i = 1; i <= n; ++i) {
cin >> a[i];
}
cin >> m;
int l, r, k;
priority_queue<int> pq;
for (int i = 0; i < m; ++i) {
cin >> l >> r >> k;
while (!pq.empty()) {
pq.pop();
}
for (int j = l; j <= r; ++j) {
pq.push(a[j]);
}
for (int j = 0; j < k-1; ++j) {
pq.pop();
}
cout << pq.top() << endl;
}
}
return 0;
}