区间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。
思路:首先输入n表示序列长度,然后输入给定序列,直接用代码表示
int[] arr=new int[n];//定义给定序列数组
for(int i=0;i<arr.lenght;i++)
{
arr[i]=cin.nextInt();
}
完成对给定序列的存储后想办法编写查找K大数的代码,题中说明给定一个m表示查询的个数,并且说明接下来有m行,且没行都是三个数i,r,k,可以定义一个m行3列的二维数组对要查询的数据进行存储
int[][] arr1=new int[m][3];
for (int i = 0; i < m; i++) {
arr1[i][0] = cin.nextInt();// i
arr1[i][1] = cin.nextInt();// r
arr1[i][2] = cin.nextInt();// k
}
因为要查询给定序列中从大往小的第k大的数,所以可以定义一个数组arr2[]用来存储给定序列中次从i到r个数
int[] arr2=new int[r-i-1];//从i到r有r-i+1个数据
for (int j=0;i<=arr2.length;j++){
arr2[j]=arr[i-1+j];//arr2数组中的第一个数据等于给定序列数组中的第i-1个数据,以此类推
}
定义一个函数来计算所要求的数字
public static int getMax(int []arr,int i,int r,int k){
int[] arr2=new int[r-i-1];
or (int j=0;i<=arr2.length;j++){
arr2[j]=arr[i-1+j];
}
Arrays.sort(arr2);//对数组进行升序排序
return arr[arr2.length-k];从大往小的第k大数就相当于从小往大排序的第arr2.length-k个数,并返回
}
在主函数中调用getMax函数,因为要查询m次所以加一个for循环
for(int i=0;i<m;i++){
System.out.println(getMax(arr[],arr1[i][0],arr1[i][1],arr[i][2]):
}
完成代码如下:
import java.util.*;
public class QuJianKDaShuChaXun {
public static void main(String[] args) {
Scanner cin = new Scanner(System.in);
int n = cin.nextInt();
int[] arr = new int[n];// 定义初始数组
// 对初始数组进行赋值
for (int i = 0; i < arr.length; i++) {
arr[i] = cin.nextInt();
}
int m = cin.nextInt();// 查询次数
int[][] arr1 = new int[m][3];// 定义二维数组存放输入的查询数据
// 对维位数组赋值
for (int i = 0; i < m; i++) {
arr1[i][0] = cin.nextInt();// i
arr1[i][1] = cin.nextInt();// r
arr1[i][2] = cin.nextInt();// k
}
for (int i = 0; i < m; i++) {
System.out.println(getMax(arr, arr1[i][0], arr1[i][1], arr1[i][2]));// 调 用getMax函数并输出
}
cin.close();
}
// 获取第K大的数
public static int getMax(int[] arr, int i, int r, int k) {
int[] arr2 = new int[r - i + 1];// 定义新数组来保存第i到第r的数据
// 对arr2数组赋值
for (int j = 0; j < arr2.length; j++) {
arr2[j] = arr[i - 1 + j];
}
Arrays.sort(arr2);// 升序排序
return arr2[arr2.length - k];// 第k大的数就是数组升序排序后倒数第k个
}
}