ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[Google] LeetCode 366 Find Leaves of Binary Tree

2022-08-20 04:30:08  阅读:151  来源: 互联网

标签:Binary Google TreeNode dep Leaves int right ans left


Given the root of a binary tree, collect a tree's nodes as if you were doing this:

  • Collect all the leaf nodes.
  • Remove all the leaf nodes.
  • Repeat until the tree is empty.

Solution

每次需要删去叶子节点。

我们利用 \(dfs\):对于叶子节点的,我们将其深度返回 \(-1\). 这样我们递归返回的时候,\(h=\max(l+1,r+1)\), 第一个下标就是 \(0\).

如果当前深度和 \(ans.size()\) 相同时,我们就 \(push\_back\) 进一个新的 \(vector\). 接着我们分别递归出 \(l\_dep, r\_dep\). 然后:

\[h=\max(l\_dep, r\_dep) \]

点击查看代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    vector<vector<int>> ans;
    
    int height(TreeNode* rt, int cur_dep){
        if(!rt) return -1;
        if(ans.size()==cur_dep){
            ans.push_back({});
        }
        int l_dep = height(rt->left, cur_dep+1);
        int r_dep = height(rt->right, cur_dep+1);
        int h = max(l_dep+1, r_dep+1);
        ans[h].push_back(rt->val);
        return h;
    }
    
public:
    vector<vector<int>> findLeaves(TreeNode* root) {
        height(root, 0);
        return ans;
    }
};



标签:Binary,Google,TreeNode,dep,Leaves,int,right,ans,left
来源: https://www.cnblogs.com/xinyu04/p/16607060.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有