尝试一
结果:答案正确:恭喜!您提交的程序通过了所有的测试用例
代码:
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
ArrayList<Integer> result = new ArrayList<>();
/* 特殊情况处理 */
if( k > input.length || k == 0){
return result;
}
/* 归并排序,为什么要传入temp,是为了避免在合并时重复创建temp.其实最后temp和input是一样的*/
int temp[] = new int[input.length];
mergeSort(0, input.length - 1, input, temp);
/* 输出k个数*/
int trueSize=k<temp.length? k:temp.length;
for (int i = 0; i < trueSize; ++i) {
result.add(temp[i]);
}
return result;
}
private void mergeSort(int left, int right, int[] input, int[] temp) {
/*递归结束条件*/
if (left == right) {
return;
}
/*分*/
int midle = (left + right) / 2;
mergeSort(left, midle, input, temp);
mergeSort(midle + 1, right, input, temp);
/* 合并(治)*/
int i = left;
int j = midle + 1;
int offset = left;
while (i <= midle && j <= right) {
if (input[i] <= input[j]) {
temp[offset++] = input[i++];
} else {
temp[offset++] = input[j++];
}
}
if (i > midle && j <= right) {
System.arraycopy(input, j, temp, offset, right - j + 1);
} else if (i <= midle && j > right) {
System.arraycopy(input, i, temp, offset, midle - i + 1);
}
/*todo 将temp拷贝会到array中*/
System.arraycopy(temp,left,input,left,right-left+1);
}
}
分析:
- 先将数组进行排序(归并排序)
- 输出前k个元素
结论
- 归并排序的合并,需要new一个数组来进行合并(还需要将这个new出来的数组重新拷回到数组中)。所以感觉还是快速排序好,归并排序麻烦了。