藍橋杯_算法訓練_區間k大數查詢

試題 算法訓練 區間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;
}