算法训练 区间k大数查询
时间限制:1.0s 内存限制:256.0MB
问题描述
给定一个序列,每次询问序列中第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
1 2 3 4 5
2
1 5 2
2 3 2
样例输出
4
2
2
数据规模与约定
对于30%的数据,n,m<=100;
对于100%的数据,n,m<=1000;
保证k<=(r-l+1),序列中的数<=106。
先创建一个数组b进行对第l个数到第k个数进行储存,然后对数组b使用排序的方法,进行从大到小进行排序,再把第K大的数值保存到输出数组里。
第一次做的代码如下:(100分)
import java.util.Arrays;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] a = new int[n];
for(int i=0;i<a.length;i++){
a[i]=scanner.nextInt();
}
int m = scanner.nextInt();
int[] b = new int[n];
int[] l = new int[m];
int[] r = new int[m];
int[] K = new int[m];
int[] result = new int[m];
for(int i=0;i<m;i++){
l[i] = scanner.nextInt();
r[i] = scanner.nextInt();
K[i] = scanner.nextInt();
}
for(int i=0;i<m;i++){
int s=0;
for(int j=l[i]-1;j<=r[i]-1;j++){
b[s]=a[j];
s++;
}
Arrays.sort(b);
result[i]=b[n-K[i]];
for(int j=0;j<n;j++){
b[j]=0;
}
}
for(int i=0;i<m;i++){
System.out.println(result[i]);
}
}
}
第二次做的代码如下:(100分)
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] a = new int[n];
for(int i=0;i<a.length;i++){
a[i]=scanner.nextInt();
}
int[] b = new int[n];
int m = scanner.nextInt();
int[] result = new int[m];
for(int i=0;i<m;i++){
int s=0;
int l = scanner.nextInt();
int r = scanner.nextInt();
int K = scanner.nextInt();
for(int j=l-1;j<=r-1;j++){
b[s]=a[j];
s++;
}
for(int j=0;j<r-l+1;j++){
for(int q=j+1;q<r-l+1;q++){
if(b[j]<b[q]){
int temp=b[j];
b[j]=b[q];
b[q]=temp;
}
}
}
result[i]=b[K-1];
}
for(int i=0;i<m;i++){
System.out.println(result[i]);
}
}
}