判断是不是完全二叉树-C++-牛客BM35

一、运行结果

二、题目

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

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

数据范围：节点数满足1≤ n≤100 

三、思路

这里采用层次遍历算法来遍历二叉树，进而判断该二叉树是否为完全二叉树。判断方法也比较简单：设置一个标识变量flag，初始为false，用于标识该二叉树是否有缺失的结点（相对于满二叉树而言的缺失），如果在层次遍历的过程中，二叉树某个位置的结点缺失，flag就设为true，由于是层次遍历，所有后遍历的结点必定在当前节点的同层或下面的层，所以，如果从缺失的结点往后还有结点，那该二叉树就不是完全二叉树（参考样例图3）；如果从第一个缺失的位置开始，后面都没有节点了，说明缺失的所有结点位于最下层的右边，该树是完全二叉树。

四、代码

class Solution {
public:
    bool isCompleteTree(TreeNode* root) {
        TreeNode* p = root;
        queue<TreeNode*> qu;
        qu.push(p);
        bool flag = false;  //表示有无缺失的结点,false表示无缺失
        while(!qu.empty()){ //层次遍历
            int num = qu.size();
            for(int i=0; i<num; i++){
                TreeNode* tmp = qu.front();
                qu.pop();
                if(tmp->left){ //左孩子结点存在
                    if(flag) return false; //前面已经有缺失的结点
                    else qu.push(tmp->left);
                }
                else flag = true;   //左孩子结点缺失
                if(tmp->right){  //右孩子结点存在
                    if(flag) return false; //前面已经有节点缺失
                    else qu.push(tmp->right);
                }
                else flag = true; //右孩子结点缺失
            }
        }
        return true;
    }
};