算法训练 区间k大数查询

150 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

问题描述

给定一个序列,每次询问序列中第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

问题分析:这题看似不好做,其实原理并不难,代码比较长而已。题目为区间k大数,则我们分为两个部分看,一部分是区间,一部分为k大数,,,根据题目要求,我们分别取输入数字为区间的开始,结束,以及第k大,,,

main中用二维数组的原因是,题目要求可输入多组数同时检测,,在main中最后一个for循环中,输出语句中调用方法,就可以输出我的调用方法中的返回值!

参考代码

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

public class Main {
    public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int[] a=new int[n];
		for(int i=0;i<n;i++){
			a[i]=sc.nextInt();
		}
		int m=sc.nextInt();
		int[][] b=new int[m][3]; 
		for(int i=0;i<m;i++){
			b[i][0]=sc.nextInt();
			b[i][1]=sc.nextInt();
			b[i][2]=sc.nextInt();
		}
		for(int i=0;i<m;i++){
		    System.out.println(f(a,b[i][0],b[i][1],b[i][2]));
	    }
    }
    public static int  f(int[] a,int r,int l,int k){
	    int[] b=new int[l-r+1];
	    for(int i=0;i<b.length;i++){
			b[i]=a[r+i-1];
		}
	    Arrays.sort(b);	
	    return b[b.length-k];
    }
}