题目:输入n数字,找出其中最小k个数字,例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4
分析:第一种方法,想到了partition函数,在位置k分割成两个部分,第一部分为小于第k个数,第二部分为大于第k个数,只用知道在k位置之前的数就是我们所求的数字。有个缺点就是改变初始数组,如果要求不改变,可以用第二种解法。
第二种方法,用到了红黑色,就是TreeSet,适合大数据。
java代码:
package LinkList;
import java.util.Iterator;
import java.util.TreeSet;
public class GetLeastNumbersMian {
// 解法一,用快速排序中的partition函数来解这个问题。但是有个问题是,partition函数会改变数组的元素的顺序
// 如果面试官要求在不改变输入的数组的情况下,可以用解法二。
public void GetLeastNumbers1(int[] input, int n, int[] output, int k) {
if (input == null || output == null || k > n || n <= 0 || k <= 0)
return;
int start = 0;
int end = n - 1;
int index = Partition(input, start, end);
while (index != k - 1) {
if (index > k - 1) {
end = index - 1;
index = Partition(input, start, end);
} else {
start = index + 1;
index = Partition(input, start, end);
}
}
for (int i = 0; i < k; i++) {
System.out.print(input[i] + " ");
output[i] = input[i];
}
}
// 解法二,使用红黑树来解决,适合海量数据。不会改变初始数组。
public void GetLeastNumbers(int[] input, TreeSet<Integer> treeset, int k) {
treeset.clear();
if (k < 1 || input.length < k)
return;
for (int i = 0; i < input.length; i++) {
if (treeset.size() < k)
treeset.add(input[i]);
else {
int a = treeset.last();
if (input[i] < a) {
treeset.remove(a);
treeset.add(input[i]);
}
}
}
Iterator<Integer> it = treeset.iterator();
while (it.hasNext()) {
System.out.print(it.next() + " ");
}
}
private int Partition(int[] a, int p, int q) {
int x, i, j, temp;
x = a[p]; // 将最后一个值保存在x中
i = p; // 开始的时候将i 移动到数组的外面
for (j = p + 1; j <= q; j++) {
if (a[j] <= x) // 比较比x大的数,并交换位置
{
i += 1;
temp = a[i]; // exchange
a[i] = a[j];
a[j] = temp;
}
}
temp = a[i]; // exchange
a[i] = a[p];
a[p] = temp;
return i;
}
public static void main(String[] args) {
GetLeastNumbersMian getLeastNumbersMian = new GetLeastNumbersMian();
int[] input = { 4, 5, 1, 6, 2, 7, 3, 8 };
int k = 4;
int output[] = new int[4];
int n = input.length;
getLeastNumbersMian.GetLeastNumbers1(input, n, output, k);
System.out.println();
TreeSet<Integer> treeset = new TreeSet<Integer>();
getLeastNumbersMian.GetLeastNumbers(input, treeset, k);
}
}
最后结果为:
3 1 2 4
1 2 3 4