算法训练 区间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)，序列中的数<=10⁶。

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

public class Algo_01_MaxSel {
    public static void main(String[] args) {
        int n, m, l, r, k;
        int array[], result[], temp[];
        Scanner scanner = new Scanner(System.in);
        
        // ***************n是序列长度
        n = scanner.nextInt();
        
        array = new int[n];

        // ***************输入序列
        for (int i = 0; i < n; i++) {
            array[i] = scanner.nextInt();
        }

        // ***************输入查询次数
        m = scanner.nextInt();
        result = new int[m];

        // ***************输入每次查询的要求，左起->右至，第k大
        for (int i = 0; i < m; i++) {

            l = scanner.nextInt();
            r = scanner.nextInt();
            k = scanner.nextInt();

            // 定义暂存数组，他的个数与区间个数相等
            int temp_i = 0;
            int length = r - l + 1;
            temp = new int[length];

            // 将区间数字一个个赋值到暂存数组里
            for (int j = (l - 1); j < (r - 1); j++) {
                temp[temp_i] = array[j];
                temp_i++;
            }

            // 对暂存数组进行排序，结果三从小到大的
            Arrays.sort(temp);
            // 将第k大的数字复制给结果数组
            result[i] = temp[length - k + 1];
        }

        // ***************分成 m 行输出查询结果
        for (int i = 0; i < m; i++) {
            System.out.println(result[i]);
        }

    }
}

 我自己的代码未通过检测，以下是网友的解题

http://blog.csdn.net/u011506951/article/details/26820363

import java.util.Arrays;  
import java.util.Scanner;  
  
public class Algo_01_MaxSel {  
  
    public static void main(String[] args) {  
        Scanner sc = new Scanner(System.in);  
        int n = sc.nextInt();  
        
        //定义一个数组num用于存放数列
        int[] num = new int[n];  
        sc.nextLine();  
        
        //输入数列
        for (int i = 0; i < n; i++) {  
            num[i] = sc.nextInt();  
        }  
  
        //输入查询的次数
        int m = sc.nextInt();  
  
        //定义存放查询指令的二维数组arr
        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++) {  
            //定义暂存数组，区间就是二维数组里，每一行的第2个数减去第1个数+1
            int[] temp = new int[arr[i][1] - arr[i][0] + 1];  
            //复制区间里的数字，注意数列是从1开始，而数组下标是0开始，所以，查询条件左右都需-1
            //temp = Arrays.copyOfRange(num, arr[i][0] - 1, arr[i][1]原作者此处忘了-1);  
            temp = Arrays.copyOfRange(num, arr[i][0] - 1, arr[i][1]-1);  
            Arrays.sort(temp);  
  
            System.out.println(temp[arr[i][1] - arr[i][0] + 1 - arr[i][2]]);  
        }  
    }  
}