试题 算法训练 区间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),序列中的数<=10^6。
代码实现:
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[] sa = new int[n];
for(int i = 0;i<n;i++)
sa[i] = sc.nextInt();
//询问次数
int m = sc.nextInt();
//创建接收序列
int[][] js = new int[m][3];
for(int i = 0;i<m;i++)
for(int j = 0;j<3;j++){
js[i][j] = sc.nextInt();
}
//算法
check(m,js,sa);
}
private static void check(int m, int[][] js, int[] sa) {
for(int i= 0;i<m;i++) {
int l = js[i][0];
int r = js[i][1];
int k = js[i][2];
//创建新数组
int[] ls = new int[r-l+1];
for(int j = l-1,s = 0;j<r;j++,s++) {
ls[s] = sa[j];
}
Arrays.sort(ls);
System.out.println(ls[ls.length-k]);
}
}
}
思路:check()方法里用新的数组接收区间的数,用Arrays.sort()函数进行简单的排序,最后注意数组位置ls[ls.length-1-k+1]