美文网首页
剑指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