美文网首页
2020-03-10 刷题1(二叉树的直径)

2020-03-10 刷题1(二叉树的直径)

作者: nowherespyfly | 来源:发表于2020-03-11 11:25 被阅读0次

543 二叉树的直径

对于根节点r,它的直径无非有三种可能:左子树的直径,右子树的直径,已经左右子树高度之和。所以后序遍历整棵树,然后得到这三个值,最后返回最大值就可以。因为c++不能返回多个值,所以我设置了一个全局变量栈,来存放左右子树的高度。
进一步简化问题,发现两棵子树的直径,归根到底还是某左右子树高度之和,而我们所需要的,也无非就是一个最大值,所以可以设置一个全局变量存放它,每次如果当前节点的左右子树高度之和比它大,就更新。

/**
 * 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:
    int max_diam = 0;
    int dep(TreeNode *root){
        if(!root) return 0;
        int left_h = dep(root -> left);
        int right_h = dep(root -> right);
        if(left_h + right_h > max_diam)
            max_diam = left_h + right_h;
        int h = left_h > right_h ? left_h + 1 : right_h + 1;
        return h;
    }
    int diameterOfBinaryTree(TreeNode* root) {
        int h = dep(root);
        return max_diam;
    }
};

相关文章

网友评论

      本文标题:2020-03-10 刷题1(二叉树的直径)

      本文链接:https://www.haomeiwen.com/subject/dawwdhtx.html