最小堆的构建

最小堆的构建

最小堆是一棵完全二叉树，每个非叶子节点的值都小于其孩子节点的值。

构建过程

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,

总结

堆能用在排序等的算法中，在此先打下基础。