LeetCode "Find Leaves of Binary Tree"

二叉树节点分层遍历

weixin_30624825

于 2016-06-27 11:35:00 发布

阅读量64

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/tonix/p/5619696.html

本文介绍了一种使用深度优先搜索(DFS)的方法来解决二叉树节点的分层遍历问题。通过递归地计算每个节点的高度，并将同一高度的节点值收集到同一数组中，最终形成按层次顺序的节点值集合。

Typical DFS problem. Simply get higher height and push_back.

/**
 * 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 {
    vector<vector<int>> ret;
    int go(TreeNode *p) // return max height
    {
        if(!p) return 0;
        
        int hl = go(p->left);
        int hr = go(p->right);
        
        int h = max(hl, hr) + 1;
        if(h > ret.size()) ret.push_back({});
        ret[h - 1].push_back(p->val);
        
        return h;
    }
public:
    vector<vector<int>> findLeaves(TreeNode* root) {
        go(root);
        return ret;
    }
};