首页 > 代码库 > [ALGO-1] 区间k大数查询

[ALGO-1] 区间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

import java.util.Arrays;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();

		int[] num = new int[n];
		sc.nextLine();
		for (int i = 0; i < n; i++) {
			num[i] = sc.nextInt();
		}

		int m = sc.nextInt();

		int[][] arr = new int[m][3];

		for (int i = 0; i < m; i++) {
			arr[i][0] = sc.nextInt();
			arr[i][1] = sc.nextInt();
			arr[i][2] = sc.nextInt();
		}

		for (int i = 0; i < m; i++) {
			int[] temp = new int[arr[i][1] - arr[i][0] + 1];
			temp = Arrays.copyOfRange(num, arr[i][0] - 1, arr[i][1]);
			Arrays.sort(temp);

			System.out.println(temp[arr[i][1] - arr[i][0] + 1 - arr[i][2]]);
		}
	}
}