Java—— 算法训练 区间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),序列中的数<=106。
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
int data1 [] = new int[1000];
int data2 [] = new int[1000];
for(int i = 0;i<n;i++) {
data1[i] = scan.nextInt();
}
int m = scan.nextInt();
for(int i = 0;i<m;i++) {
int l = scan.nextInt();
int r = scan.nextInt();
int K = scan.nextInt();
int count = 0;
for(int j = l-1;j<r;j++) {
data2[count ++] =data1[j];
}
Arrays.sort(data2, 0, count);
System.out.println(data2[count - K]);
}
}
}
下面是用容器写的,但是结果为超时??不是很明白,数组排序和容器List里面的排序有何不同??
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
List<Integer> list = new ArrayList<Integer>();
List<Integer> rlist = new ArrayList<Integer>();
for(int i = 0;i<n;i++) {
list.add(i, scan.nextInt());
}
int m = scan.nextInt();
for(int i = 0;i<m;i++) {
int l = scan.nextInt();
int r = scan.nextInt();
int K = scan.nextInt();
for(int j = l - 1; j < r;j++) {
rlist.add(list.get(j));
}
rlist.sort(null);
System.out.println(rlist.get(rlist.size() - K));
rlist.clear();
}
}
}
容器结果:
数组结果: