蓝桥杯 算法训练 区间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)，序列中的数<=10⁶。

思路

这题是一道简单的排序加搜索问题

刚拿到这题我没注意数据范围，想用ArrayList做的，但是发现数据不好复制，所以更换用数组做

排序因为效率问题使用了时间复杂度为O(nlogn)的归并排序

代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
    	Scanner scan=new Scanner(System.in);
    	int a[]=new int[1001];
    	int n=scan.nextInt();
    	for(int i=0;i<n;i++)
    	{
    		a[i]=scan.nextInt();
    	}
    	int[] temp=new int[1001];
    	int[] b=new int[1001];
    	int t=scan.nextInt();
    	for(int i=0;i<t;i++)
    	{
    		int begin=scan.nextInt();
    		int end=scan.nextInt();
    		int which=scan.nextInt();
    		for(int j=begin-1;j<end;j++)
		    	{
    			 	b[j]=a[j];
		    	}
    		mergeSort(b,temp,begin-1,end-1);
    		System.out.println(b[end-which]);
    	}
    }
    private static void mergeSort(int b[],int temp[],int left,int right)
    {
    	if(left<right)
    	{
    		int mid=(left+right)/2;
    		mergeSort(b,temp,left,mid);
    		mergeSort(b,temp,mid+1,right);
    		merge(b,temp,left,mid,right);
    	}
    }
    private static void merge(int b[],int temp[],int left,int leftEnd,int rightEnd)
    {
    	int right=leftEnd+1;
    	int len=rightEnd-left+1;
    	int tempPos=left;
    	
    	while(left<=leftEnd&&right<=rightEnd) {
    		if(b[left]<=b[right]) {
    			temp[tempPos++]=b[left++];
    		}
    		else
    			temp[tempPos++]=b[right++];
    	}
    	while(left<=leftEnd)
    	{
    		temp[tempPos++]=b[left++];
    	}
    	while(right<=rightEnd)
    	{
    		temp[tempPos++]=b[right++];
    	}
    	for(int i=0;i<len;i++,rightEnd--)
    	{
    		b[rightEnd]=temp[rightEnd];
    	}
    }
}

这里有个小坑

对于第一次读入所有数据的数组a是不能改变其数据顺序的