java 最小数_java_最小的k个数

package com.zhiru;

/*

* 求一个序列中最小的k个数

*/

public class TopK {

private int[] heap;

private int currentSize;

TopK(int[] a) {

if (a != null) {

heap = new int[a.length];

currentSize = 0;

for (int i = 0; i < a.length; i++)

heap[currentSize++] = a[i];

}

heapAdjust();

}

//整体自底向上调整，局部自顶向下调整。

public void heapAdjust() {

int currentPos = (currentSize - 2) / 2;

while (currentPos >= 0) {

siftDown(currentPos, currentSize - 1);

currentPos--;

}

}

// 最小堆局部调整算法

// 假定前提是子女已是成最小堆.

private void siftDown(int start, int m) {

int i = start;

int j = 2 * i + 1;// 开始节点的左子女。

int temp = heap[i];

while (j <= m) {

if (j < m && heap[j] > heap[j + 1])

j++;// j指向俩子女中的小者

if (temp <= heap[j])

break;// 父节点小不做调整。

else {

heap[i] = heap[j];// 关键码小的子女结点上浮.

i = j;

j = 2 * i + 1;// j下移

}

}

heap[i] = temp;

}

/*

* a是输入的序列 函数实现获取最小的k个元素.

*/

public void getTopK(int[] a, int k) {

for (int j = currentSize - 1; j >= 0; j--, k--) {

if (k > 0) {

System.out.println(heap[0]);

int temp = heap[0];

heap[0] = heap[j];

heap[j] = temp;

siftDown(0, j - 1);

}

else

break;

}

}

public static void main(String[] args) {

// TODO Auto-generated method stub

int[] a = { 8, 2, 5, 4, 3, 6, 9, 0, 7, 100 };

TopK tk = new TopK(a);

tk.getTopK(a, 6);

}

}0 2 3 4 5 6