Leetcode 958. 二叉树的完全性检验

JehanRio

已于 2023-10-01 23:44:29 修改

阅读量385

点赞数 1

分类专栏： Leetcode专栏文章标签： leetcode 算法职场和发展

于 2023-08-17 09:41:47 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/132334043

版权

Leetcode专栏专栏收录该内容

278 篇文章 3 订阅

订阅专栏

文章目录

题目
代码（8.17 首刷看解析）
代码（10.1 二刷自解）

题目

在这里插入图片描述
Leetcode 958. 二叉树的完全性检验

代码（8.17 首刷看解析）

这个方式更简单，将nil也加入进去，只能有一个nil，如果后面再出现nil就直接false了

func isCompleteTree(root *TreeNode) bool {
    q := []*TreeNode{root}
    flag := false
    for len(q) > 0 {
        front := q[0]
        q = q[1:]
        if front == nil {
            flag = true
            continue
        }
        if flag == true {
            return false
        }
        q = append(q, front.Left)
        q = append(q, front.Right)
    }
    return true
}

class Solution {
public:
    bool isCompleteTree(TreeNode* root) {
        queue<TreeNode*> q;
        q.push(root);
        bool flag = false;
        while(!q.empty()) {
            auto front = q.front();
            q.pop();
            if(!front->left && front->right)
                return false;
            if(flag && (front->left || front->right))
                return false;
            if((!front->left && !front->right) || (front->left && !front->right))
                flag = true;
            if(front->left) q.push(front->left);
            if(front->right) q.push(front->right);
        }
        return true;
    }
};

代码（10.1 二刷自解）

class Solution {
public:
    bool isCompleteTree(TreeNode* root) {
        queue<TreeNode*> q;
        q.push(root);
        bool flag = false;
        while(!q.empty()) {
            int sz = q.size();
            
            for(int i = 0; i < sz; i++) {
                auto front = q.front();
                q.pop();
                if(flag && (front->left || front->right)) return false;
                if(front->left) {
                    if(flag)
                        return false;
                    q.push(front->left);
                } else 
                    flag = true;
                if(front->right) {
                    if(flag)
                        return false;
                    q.push(front->right);
                }   
                else flag = true;
            }
        }
        return true;
    }
};