蓝桥杯 算法训练 区间k大数查询 java

记录 2018 7

问题描述

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

思路：

这题使用快排是一个比较好的做法，快排的原理就是找到任意一个数，把小于它的放左边，大于它的放右边，所以如果把任意数改为你要的k大数呢？不过此题比较简单，只需要调用内置排序方法即可，因为sort是从小到大排的，所以倒数第一个是最大数，下标r-l+1-k即是k大数

 

import java.util.Arrays;
import java.util.Scanner;
public class Main {

	public static void main(String[] args) {
		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();
		int[] ans = new int[m];
		for(int i=0;i<m;i++)
		{
			int l = in.nextInt();
			int r = in.nextInt();
			int k = in.nextInt();
			int[] b = new int[r-l+1];
			for(int j=0;j<r-l+1;j++)
			{
				b[j]=a[j+l-1];
			}
			Arrays.sort(b);
			ans[i]=b[r-l+1-k];
		}
		for(int i=0;i<m;i++)
		{
			System.out.println(ans[i]);
		}
		

	}

}