LeetCode543

二叉树的直径

分析

开始简单的以为其实就是左右最大深度之和，然后发现有点不对劲，其实如果这个树很不平衡，那么这个最大直径极有可能出现在某颗子树上。所以正解其实有很多种写法，可以向像我这样直接无脑的写

/**
 * 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 ans=0;
    int diameterOfBinaryTree(TreeNode* root) {
        if(root==NULL)return 0;
        solve(root);
        return ans;
    }
    void solve(TreeNode * root){
        if(root==NULL)return ;
        int lm=getDep(root->left);
        int rm=getDep(root->right);
        ans=max(ans,lm+rm);
        solve(root->left);
        solve(root->right);
    }
    int getDep(TreeNode *rt){
        if(rt==NULL)return 0;
        return max(getDep(rt->left),getDep(rt->right))+1;
    }
};

其他解法其实还有很多，我看看是不是明天有空来补一个bfs的写法，就是正规求直径的写法。或者枚举叶子节点LCA的写法，以及记忆化DP，大概是有很多写法。