【leetcode_easy】543. Diameter of Binary Tree

problem

543. Diameter of Binary Tree

题意:

转换一种角度来看,是不是其实就是根结点1的左右两个子树的深度之和呢。那么我们只要对每一个结点求出其左右子树深度之和,这个值作为一个候选值,然后再对左右子结点分别调用求直径对递归函数,这三个值相互比较,取最大的值更新结果res,因为直径不一定会经过根结点,所以才要对左右子结点再分别算一次。为了减少重复计算,我们用哈希表建立每个结点和其深度之间的映射,这样某个结点的深度之前计算过了,就不用再次计算了。

solution1: 两个递归函数。

/**
 * 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 diameterOfBinaryTree(TreeNode* root) {
        if(root==NULL) return 0;
        int ans = getHeight(root->left) + getHeight(root->right);
        return max(ans, max(diameterOfBinaryTree(root->left), diameterOfBinaryTree(root->right)));
    }
    int getHeight(TreeNode* node)
    {
        if(node==NULL) return 0;
        if(mymap.count(node)) return mymap[node];
        int h = 1 + max(getHeight(node->left), getHeight(node->right));
        mymap[node] = h;
        return h;
    }
private:
    unordered_map<TreeNode*, int> mymap;
};

solution2: 一个递归函数。

只需要用一个递归函数就可以了,可以在求深度的递归函数中顺便就把直径算出来,而且貌似不用进行优化也能accept。

 

solution3: 

 优化版本。

 

参考

1. Leetcode_easy_543. Diameter of Binary Tree;

2. Grandyang;

posted on 2019-05-29 13:08  鹅要长大  阅读(209)  评论(0编辑  收藏  举报

导航