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

跟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> levelOrder(TreeNode* root) {
        vector> result={};
        if (root==NULL)
            return result;
        queueque;
        que.push(root);
        que.push(NULL);
        vectorres={};
        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 二叉树分层)