算法训练 区间k大数查询

2024年2月11日 172次阅读

题目

问题描述

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

思路：

幸好数据规模小~

代码

import java.util.*;
import java.math.*;
import java.util.regex.*;


public class Main{

    final static int INF = 0x3f3f3f3f;
    final static int NUM = 100;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int n,m,l,r,K;
        while(sc.hasNext()) {
            n=sc.nextInt();
            int[] a = new int[n];
            for(int i=0;i<a.length;i++)a[i]=sc.nextInt();
            m=sc.nextInt();
            for(int i=0;i<m;i++) {
                l=sc.nextInt();r=sc.nextInt();K=sc.nextInt();
                int[] b = new int[r-l+1];
                for(int j=l,k=0;j<=r;j++,k++)
                    b[k]=a[j-1];
                Arrays.sort(b);
                System.out.println(b[r-l+1-K]);
            }
        }

    }
}
点赞