蓝桥杯练习-算法训练-区间k大数查询

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