美文网首页
剑指32-2/leetcode102 二叉树分层

剑指32-2/leetcode102 二叉树分层

作者: vaisy | 来源:发表于2022-03-31 17:40 被阅读0次

    跟32的区别是,每一层的要单独进一个数组。所以需要一个flag位置来判断是否这层已经结束了。
    flag位置的加法:上一层最后一个点进去的时候,加完左右子树加入flag
    1
    2 3
    45 67

    先读root,1进队之后输入null,2,3进队。
    1出队遇到null,于是2,3后加入null;
    2出队,45进队;3出队遇到null,于是67进队,加入null
    32-3类似,在指定层的时候 reverse一下就可以。

    class Solution {
    public:
        vector<vector<int>> levelOrder(TreeNode* root) {
            vector<vector<int>> result={};
            if (root==NULL)
                return result;
            queue<TreeNode*>que;
            que.push(root);
            que.push(NULL);
            vector<int >res={};
            while(!que.empty()){
                TreeNode* x=que.front();
                que.pop();
                if(x==NULL){
                    if(res.size()){
                        result.push_back(res);
                        res={};
                        que.push(NULL);
                    }else break;
                    continue;
                }
                res.push_back(x->val);
                if(x->left)
                    que.push(x->left);
                if(x->right)
                    que.push(x->right);
            }
            return result;
        }
    };
    

    相关文章

      网友评论

          本文标题:剑指32-2/leetcode102 二叉树分层

          本文链接:https://www.haomeiwen.com/subject/mkzgjrtx.html