蓝桥杯——算法训练——ALGO-1——区间k大数查询

我的蓝桥杯代码仓：https://github.com/617076674/lanqiao

题目描述：

知识点：递归

思路：递归地求解无序集合中的第K大数

取索引为0的元素作为标准standard将原集合划分成2个集合，一个集合greaterList，其中的元素均大于standard，另一个集合lessList，其中的元素均小于standard。

如果greaterList中的元素个数是K - 1，说明standard就是第K大数。

如果greaterList中的元素个数小于K - 1，在lessList中递归地寻找第K - 1 - greaterList.size()大数。

如果greaterList中的元素个数大于K - 1，在greaterList中递归地寻找第K大数。

时间复杂度和空间复杂度均与给定的查询区间[l, r]的大小有关，对于单词查询，时间复杂度和空间复杂度是O(log(r - l))。

JAVA代码：

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		Integer[] nums = new Integer[n];
		for(int i = 0; i < n; i++) {
			nums[i] = scanner.nextInt();
		}
		int m = scanner.nextInt();
		for(int i = 0; i < m; i++) {
			int l = scanner.nextInt();
			int r = scanner.nextInt();
			int K = scanner.nextInt();
			ArrayList<Integer> arrayList = new ArrayList<Integer>();
			for(int j = l - 1; j <= r - 1; j++) {
				arrayList.add(nums[j]);
			}
			int result = findKMax(arrayList, K);
			System.out.println(result);
		}
	}
	
	private static int findKMax(ArrayList<Integer> arrayList, int K) {
		int standard = arrayList.get(0);
		ArrayList<Integer> greaterList = new ArrayList<Integer>();
		ArrayList<Integer> lessList = new ArrayList<Integer>();
		for(int i = 1; i < arrayList.size(); i++) {
			int num = arrayList.get(i);
			if(num >= standard) {
				greaterList.add(num);
			}else {
				lessList.add(num);
			}
		}
		if(greaterList.size() == K - 1) {
			return standard;
		}else if(greaterList.size() < K - 1) {
			return findKMax(lessList, K - 1 - greaterList.size());
		}else {
			return findKMax(greaterList, K);
		}
	}
}

JAVA解题报告：