LeetCode题组：第543题-二叉树的直径

1.题目

难度：简单

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

示例 :

给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

---

2.我的解答

把问题分解为： 直径 = max{目前最大直径，左子树深度 + 右子树深度}

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int max (int a,int b) {
    return a > b ? a : b;
}
int travelTree(struct TreeNode * root, int * diameter) {
    if (root == NULL) {
        return 0;
    }
    int left = travelTree(root->left, diameter);
    int right = travelTree(root->right, diameter);
    *diameter = max(*diameter, left + right);
    return 1 + max(left, right);
}

int diameterOfBinaryTree(struct TreeNode* root){
    int diameter = 0;
    travelTree(root, &diameter);
    return diameter;
}