藍橋杯算法訓練 區間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);
    }
}