蓝桥杯 区间K大数查询(java题解)

算法训练 区间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。

题解：

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        if(n>1000)
            System.exit(0);
        int[] a = new int[n];
        for(int i=0;i<n;i++) {
            int q = in.nextInt();
            if(q>1000000)
                System.exit(0);
            a[i] = q;


        }
        int m = in.nextInt();
        if(m>1000)
            System.exit(0);

        int[] l = new int[m];
        int[] r = new int[m];
        int[] K = new int[m];

        for(int i=0;i<m;i++) {
            l[i] = in.nextInt();
            r[i] = in.nextInt();
            K[i] = in.nextInt();
            if(K[i]>(r[i]-l[i]+1)) 
                System.exit(0);

        }

        int[] res = new int[m];
        for(int i=0;i<m;i++) {
            int[] b = new int[r[i]-l[i]+1];
            int w = l[i] - 1;
            for(int j=0;j<b.length;j++) {
                b[j] = a[w];
                w++;
            }
            Arrays.sort(b);
            res[i] = b[b.length-K[i]];
        }
        for(int i=0;i<m;i++) {
            System.out.println(res[i]);
        }

        in.close();
    }
}