二叉树的层序遍历II

题目描述

给定一个二叉树，返回其节点值自底向上的层序遍历。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）

输入输出样例

给定一个二叉树 [3,9,20,null,null,15,7]

  3
 / \
9  20
  /  \
 15   7

返回其自底向上的层序遍历为：

[
  [15,7],
  [9,20],
  [3]
]

题解

以二叉树的层序遍历作为蓝本，我们仅需将其作一点的改动，就可以写出结果了

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        queue<TreeNode*> res;
        vector<vector<int>> ans;
        if (root != nullptr) res.push(root);

        while (!res.empty()) {
            int size = res.size();
            vector<int> vec;
            for (int i = 0;i < size; i++) {
                TreeNode* node = res.front();
                res.pop();
                vec.push_back(node->val);
                if (node->left) res.push(node->left);
                if (node->right) res.push(node->right);
            }
            ans.push_back(vec);
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
};