藍橋杯練習-演算法訓練-區間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;
}