Problem Description
还记得Gardon给小希布置的那个作业么?(上次比赛的1005)其实小希已经找回了原来的那张数表,现在她想确认一下她的答案是否正确,但是整个的答案是很庞大的表,小希只想让你把答案中最大的M个数告诉她就可以了。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。
Input
输入可能包含多组数据,其中每组数据包括两行:
第一行两个数N和M,
第二行N个数,表示该序列。
第一行两个数N和M,
第二行N个数,表示该序列。
Output
对于输入的每组数据,输出M个数,表示结果。输出应当按照从大到小的顺序排列。
Sample Input
4 4
1 2 3 4
Sample Output
7 6 5 5
这就是题目,开始我利用快速排序找第K大的元素,然后将前面的元素再快速排序,以为这样会快。。。感觉直接排序会慢很多。。。代码是:
import java.util.Arrays;
import java.util.Scanner;
public class HDOJ1280 {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
while(scan.hasNext()){
int n = scan.nextInt();
int k = scan.nextInt();
int a[] = new int [n+1];
int b[] = new int [k+1];
int x = n*(n-1)/2;
int c [] = new int [x+1];
for(int i = 1;i<=n;i++){
a[i] = scan.nextInt();
}
int location = 1;
for(int i = 1 ; i<=n;i++){
for(int j = i+1 ; j<=n;j++){
c[location] = a[i] + a[j];
location++;
}
}
new HDOJ1280().findKth(c, b, k, 1, x);
new HDOJ1280().quick(c, 1, k);
for(int i = 1 ; i<= k ; i ++){
System.out.print(c[i]+" ");
}
System.out.println();
}
}
public int partition(int a[], int f, int r) {
int point = f;
int i = f + 1;
int j = r;
int p = a[point];
while (true) {
while (i <=r && a[i] > p) {
i++;
}
while (j > 0 && a[j] < p) {
j--;
}
if (j <= i) {
break;
} else {
int x = a[j];
a[j] = a[i];
a[i] = x;
}
}
a[point] = a[j];
a[j] = p;
return j;
}
public void quick(int a[] , int f, int r){//快速排序
if(f<r){
int p = partition(a,f,r);
quick(a,f,p-1);
quick(a,p+1,r);
}
}
public void findKth(int a[], int b[], int k, int f, int r) {//查找第K大的元素后返回,就是前面是前K大元素
if (f < r) {
int p = partition(a, f, r);
if (p == k) {
return ;
} else {
if (p < k) {
findKth(a, b, k, p + 1, r);
} else {
findKth(a, b, k, f, p - 1);
}
}
}
return ;
}
}
但是提交后
Time Limit Exceeded, 后来改用Java里面Arrays.sort,通过了,而且代码仅用了1/3,
import java.util.Arrays;
import java.util.Scanner;
public class test {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
while (scan.hasNext()) {
int n = scan.nextInt();
int k = scan.nextInt();
int a[] = new int[n];
int x = n * (n - 1) / 2;
int c[] = new int[x];
for (int i = 0; i < n; i++) {
a[i] = scan.nextInt();
}
int location = 0;
for (int i = 0; i <n; i++) {
for (int j = i + 1; j < n; j++) {
c[location] = a[i] + a[j];
location++;
}
}
Arrays.sort(c);
for (int i = x-1; i>x-1-k; i--) {
System.out.print(c[i]);
if(i!=x-k)
System.out.print(" ");
}
System.out.println();
}
}
}
感觉奇怪,看了Java API才发现,原来sort方法就是利用快速排序的。应该是数据<=3000,我的算法不能体现优点,当数据非常多时,感觉我的就会比Arrays.sort快了。紧紧是感觉,我也不能输入几千个数来测试。。,