leetcode 958.二叉树的完全性检验 Java

题目链接

https://leetcode-cn.com/problems/check-completeness-of-a-binary-tree/

描述

给定一个二叉树，确定它是否是一个完全二叉树。

百度百科中对完全二叉树的定义如下：

若设二叉树的深度为 h，除第 h 层外，其它各层 (1～h-1) 的结点数都达到最大个数，第 h 层所有的结点都连续集中在最左边，这就是完全二叉树。（注：第 h 层可能包含 1~ 2h 个节点。）

提示：

树中将会有 1 到 100 个结点。

示例

示例 1：

输入：[1,2,3,4,5,6]
输出：true
解释：最后一层前的每一层都是满的（即，结点值为 {1} 和 {2,3} 的两层），且最后一层中的所有结点（{4,5,6}）都尽可能地向左。

示例 2：

输入：[1,2,3,4,5,null,7]
输出：false
解释：值为 7 的结点没有尽可能靠向左侧。

初始代码模板

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isCompleteTree(TreeNode root) {

    }
}

代码

注意检测位置就好了，依旧是层序遍历的思路

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isCompleteTree(TreeNode root) {
        int size = 0;
        Queue<TreeNode> nodeQueue = new LinkedList<>();
        Queue<Integer> posQueue = new LinkedList<>();
        nodeQueue.offer(root);
        posQueue.offer(0);

        while (!nodeQueue.isEmpty()) {
            for (int num = nodeQueue.size(); num > 0; num--) {
                TreeNode node = nodeQueue.poll();
                int pos = posQueue.poll();

                if (node.left != null) {
                    size++;
                    int cur = pos * 2 + 1;
                    if (cur != size) {
                        return false;
                    }
                    nodeQueue.offer(node.left);
                    posQueue.offer(cur);
                }

                if (node.right != null) {
                    size++;
                    int cur = pos * 2 + 2;
                    if (cur != size) {
                        return false;
                    }
                    nodeQueue.offer(node.right);
                    posQueue.offer(cur);
                }
            }
        }

        return true;
    }
}