【LeetCodes刷题】-C++-简单-107-二叉树的层次遍历II

Task:

【LeetCodes刷题】-C++-简单-107-二叉树的层次遍历II_第1张图片

思路与代码:

还是用递归的思想,要读懂题目,反向输出的是一个int类型二维数组,所以就是vector>
那么思路就出来了:
(1)创建二维数组:
(2)往数组里添加树的值(正向添加);
(3)对创建的二维数组反向排序然后输出;

这里重要的步骤就是第二步,
就需要遍历树,这里设置一个level的变量,用来记录遍历到树的第几层,也就是树的深度,

  • 当树节点是空的,那直接返回就好了;
  • 如果树非空,那就把先给二位数字初始化第一个元素(就是在该二维数组中初始化一个一维数组,此时结果的二维数组可以理解为[[],]
  • 然后在这个二维数组中的第一个元素中添加当前层的元素res[level].push_back(root->val)
  • 接下来就是递归,不断的调用levelorder函数本身,给根节点的左孩子和右孩子分别进行以上步骤的操作,最后就得到这样的形式[[],[],[]...]
    这样就完成了第二步骤,得到的是一个从树的根到叶的二维数组,
    然后第三部反向输出,就好了!

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector> levelOrderBottom(TreeNode* root) {
        vector> res;
        levelorder(root, 0, res);
        return vector> (res.rbegin(), res.rend());
    }
    void levelorder(TreeNode* root, int level, vector> &res){
        if( !root ) return;
        if(res.size() == level) res.push_back({});
        res[level].push_back(root->val);
        if(root->left) levelorder(root->left, level+1, res);
        if(root->right) levelorder(root->right, level+1, res);
    }
};

提交结果:
【LeetCodes刷题】-C++-简单-107-二叉树的层次遍历II_第2张图片

你可能感兴趣的:(C/C++学习,leetCode,LeetCode刷题系列)