蓝桥杯-算法训练 区间k大数查询（排序 查找）

问题描述

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

package lanqiaobei;

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

public class Main {
	//一趟快速排序
	public static int partition(int[]a,int low,int high)
	{
		int i=low,j=high,pivot=a[low];
		while(i<j)
		{
			while(i<j&&a[j]<=pivot)
			{
				j--;
			}
			if(i<j)
			{
				a[i]=a[j];
				i++;
			}		
			while(i<j&&a[i]>=pivot)
			{
				i++;
			}
			if(i<j)
			{
				a[j]=a[i];
				j--;
			}		
		}
		a[i]=pivot;
		return i;
	}
	
	//进行降序排序
	public static void find(int[]a,int low,int high,int k)
	{
		if(low<high)
		{
			int pivotloc=partition(a,low,high);
			int len=pivotloc-low+1;
			if(len<k)
			{
				find(a,pivotloc+1,high,k-len);
			}
			else if(len>k)
			{
				find(a,low,pivotloc-1,k);
			}
		}
	}
	
    public static void main(String[] args) {
    	Scanner sc=new Scanner(System.in);
    	int n=sc.nextInt();
    	int []a=new int[n];
    	for(int i=0;i<n;i++)
    	{
    		a[i]=sc.nextInt();
    	}
    	int m=sc.nextInt();
    	int []c=new int[m];
    	for(int j=0;j<m;j++)
    	{
    		int l=sc.nextInt();
        	int r=sc.nextInt();
        	int k=sc.nextInt();
        	int []b=Arrays.copyOf(a, a.length);//将数组a的元素copy到b数组中,保证原数组不变
        	find(b,l-1,r-1,k);
        	c[j]=b[l-1+k-1];
    	}
    	for(int i=0;i<m;i++)
    	{
    		System.out.println(c[i]);
    	}
    }
}

在这里，我是直接把数组a的所有元素copy到数组b当中的，所以所占的空间会比较大。我觉得还需要改进一下，就是在每一次询问当中，只需把需要询问的元素放到一个新数组里面，然后进行排序输出第k大的数值。