蓝桥杯算法训练 区间k大数查询（Java解法）

问题描述

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

线性选择算法实现

import java.util.Scanner;

public class Main {
    
    
    private static Scanner scanner = new Scanner(System.in);
    private static int n,m;
    private static int[] nums;
    private static int l,r,k;

    public static void main(String[] args) {
    
    
        n = scanner.nextInt();
        nums = new int[n];
        for (int i = 0; i < n; i++) {
    
    
            nums[i] = scanner.nextInt();
        }
        m = scanner.nextInt();
        for (int i = 0; i < m; i++) {
    
    
            l = scanner.nextInt();
            r = scanner.nextInt();
            k = scanner.nextInt();
            int[] numsTemp = new int[r - l + 1];
            for (int j = 0; j < r-l+1; j++) {
    
    
                numsTemp[j] = nums[l-1+j];
            }
            System.out.println(lineSearch(numsTemp,0,r-l,k));
        }
    }

    /**
     * 线性从大到小第几个
     * @param arr
     */
    private static int lineSearch(int[] arr,int left,int right,int k){
    
    
        if (left >= right) return arr[left];
        int point = arr[left];
        int i = left;
        int j = right;
        while (i < j) {
    
    
            while (i < j && arr[j] <= point) j--;
            arr[i] = arr[j];
            while (i < j && arr[i] >= point) i++;
            arr[j] = arr[i];
        }
        if(j - left + 1 == k) return point;
        arr[j] = point;
        if (j - left + 1 < k) return lineSearch(arr,j+1,right,k-j-1+left);
        return lineSearch(arr,left,j-1,k);
    }
}