算法训练 区间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。

---------------------------------------------------------------------------------------------

一开始自己上传时，却超时；主要原因在于采用冒泡排序，花费的时间复杂度为n2；然后改为用直接插入排序法，时间复杂度大大降低，顺利通过编译。

import java.util.Scanner;

public class Main{
	public static void main(String[] args)
	{
		Scanner reader=new Scanner(System.in);
		int n=reader.nextInt();
		int[] array=new int[n];
		for(int i=0;i<n;i++)
		{
			array[i]=reader.nextInt();
		}
		int m=reader.nextInt();
		int[] l=new int[m];
		int[] r=new int[m];
		int[] K=new int[m];
		int[][] sort=new int[m][];
		for(int i=0;i<m;i++)
		{
			l[i]=reader.nextInt()-1;
			r[i]=reader.nextInt()-1;
			K[i]=reader.nextInt();
		}
		for(int i=0;i<m;i++)
		{
			sort[i]=new int[r[i]-l[i]+1];
			int x=0;
			for(int s=l[i];s<=r[i];s++)
			{
				sort[i][x++]=array[s];
			}
		}
		for(int i=0;i<m;i++)
		{
			int max;
			int temp;
			for(int s=0;s<K[i];s++)
			{
				max=s;
				for(int x=s;x<sort[i].length;x++)
				{
					if(sort[i][max]<sort[i][x])
						max=x;
				}
				temp=sort[i][s];
				sort[i][s]=sort[i][max];
				sort[i][max]=temp;
			}
		}
		for(int i=0;i<m;i++)
		{
		System.out.println(sort[i][K[i]-1]);
		}

	}
}