判断一棵树是否是完全二叉树

2019-11-27  本文已影响0人  Ramsey16k

按层遍历,存在两种情况:
(1)当前节点有右孩子却没有左孩子,一定不是完全二叉树
(2)当前节点不是同时有左右两个孩子的时候,后面遍历到的所有节点都必须是叶节点,否则不是完全二叉树

具体方法在注释写的很清楚,代码如下:

public class IsCBT {
    public static class Node {
        public int value;
        public Node left;
        public Node right;

        public Node(int data) {
            this.value = data;
        }
    }

    public static boolean isCBT(Node head) {
        if (head == null) {
            return true;
        }
        Queue<Node> queue = new LinkedList<>();
        boolean leaf = false; // 是否进入判断叶子节点的阶段
        Node l;
        Node r;
        queue.offer(head);
        while (!queue.isEmpty()) {
            head = queue.poll(); // 从队列里弹出当前节点
            l = head.left;
            r = head.right;
            if ((leaf && (l != null || r != null)) // 开启了判断叶子节点的阶段,且当前节点不是叶子节点(存在左、右孩子)
                    ||
                    (l == null && r != null)) { // 左孩子为空,右孩子不为空
                return false;
            }
            // 既然是按层遍历,就应该先加左孩子,再加右孩子
            if (l != null) {
                queue.offer(l);
            }
            if (r != null) {
                queue.offer(r);
            } else { // 左孩子 或 右孩子为空
                leaf = true; // 进入判断叶子节点的阶段
            }
        }
        return true;
    }

    public static void main(String[] args) {
        Node head = new Node(4);
        head.left = new Node(2);
        head.right = new Node(6);
        head.left.left = new Node(1);
        head.left.right = new Node(3);
        head.right.left = new Node(5);

        System.out.println(isCBT(head));

    }
}
上一篇 下一篇

猜你喜欢

热点阅读