最小堆的构建
最小堆是一棵完全二叉树,每个非叶子节点的值都小于其孩子节点的值。
构建过程
1、创建POJO类定义树节点来保存数据。
class Node {
private Node left;
private Node right;
private Integer value;
public Node(Node left, Node right, Integer value) {
this.left = left;
this.right = right;
this.value = value;
}
public Node getLeft() {
return left;
}
public Node getRight() {
return right;
}
public Integer getValue() {
return value;
}
public void setValue(Integer value) {
this.value = value;
}
@Override
public String toString() {
return value + "";
}
}
2、给二叉树填充数据
public static Integer[] in = {5, 1, 13, 3, 16, 7, 10, 14, 6};//原始数据
public static Node[] dd = new Node[in.length];//用数组保存所有树节点
public void buildTree(){//给所有节点赋值
int len = in.length;
for (int i = len - 1; i >= 0; i--) {//从原始数据的尾部向头部遍历
Node ld = null;
Node rd = null;
if (2 * i + 2 < len) {
rd = dd[2 * i + 2];
}
if (2 * i + 1 < len) {
ld = dd[2 * i + 1];
}
Node thisNode = new Node(ld, rd, in[i]);
dd[i] = thisNode;
}
}
对于任意非叶子节点,与其子节点的值进行比较,如果子节点的值小于它,则进行值的交换,然后对各自的子节点进行递归比较大小。
public void turnNode(Node node) {//根据节点的值进行比较和交换。
Node left = node.getLeft();
Node right = node.getRight();
if (null != right) {
if (right.getValue() < node.getValue()) {
Integer rightValue = right.getValue();
right.setValue(node.getValue());
node.setValue(rightValue);
turnNode(right);
}
if (left.getValue() < node.getValue()) {
Integer leftValue = left.getValue();
left.setValue(node.getValue());
node.setValue(leftValue);
turnNode(left);
}
} else if (null != left && left.getValue() < node.getValue()) {
Integer leftValue = left.getValue();
left.setValue(node.getValue());
node.setValue(leftValue);
}
}
测试用例
@Test
public void minHeap() {
buildTree();
for (int i = dd.length - 1; i >= 0; i--) {
turnNode(dd[i]);
}
System.out.print("最小堆为:");
for (int i = 0; i < dd.length; i++) {
System.out.print(dd[i] + ",");
}
}
程序运行结果
最小堆为:1,3,7,5,16,10,13,14,6,
总结
堆能用在排序等的算法中,在此先打下基础。