算法训练 区间k大数查询

Posted 2020-10-22 zdy1996

问题描述

给定一个序列，每次询问序列中第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.io.BufferedReader;  
import java.io.IOException;  
import java.io.InputStreamReader;
import java.text.DecimalFormat;
import java.util.*;
public class Main {  
  public static void main(String[] args) throws IOException{  
	    
	  		Scanner in= new Scanner(System.in);
	  		int n=in.nextInt();
	  		
	  		int []a=new int[n];
	  		for(int i=0;i<n;i++)
	  		{
	  			a[i]=in.nextInt();
	  		}
	  		int m=in.nextInt();
	  		
	  		while(m>0)
	  		{
	  			int a1=in.nextInt();
	  			int a2=in.nextInt();
	  			int k=in.nextInt();
	  			int length=a2-a1+1;
	  			
	  			int[]b =new int[length];
	  			
	  			for(int i=a1;i<=a2;i++)
	  			{
	  				b[i-a1]=a[i-1];
	  			}
	  			Arrays.sort(b);
	  			
	  			System.out.println(b[length-k]);
	  			m--;
	  		}
  }
}