Leetcode 543 : 二叉树两节点间拥有的最大距离

算法题，二叉树两节点间拥有的最大距离

Leetcode 543

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

class Solution {
public:
    int diameterOfBinaryTree(TreeNode* root) {
        inDiameter = 0;
        search(root);
        return inDiameter;
    }
private:
    int search(TreeNode* root) {
        if(root == NULL)
            return 0;
        int leftMax = search(root->left);
        int rightMax = search(root->right);
        if(leftMax + rightMax > inDiameter)
            inDiameter = leftMax + rightMax;
        return max(leftMax,rightMax)  + 1;
    }   
private:
    int inDiameter;
};