算法训练 区间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
;
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]);
}

}
}