Leetcode 543 - 二叉树的直径

题目

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

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    

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

注意：两结点之间的路径长度是以它们之间边的数目表示。

思路

author's blog == http://www.cnblogs.com/toulanboy/

考核：树的遍历、树的深度。

思路：*最长的路径必然经过某个点，枚举经过某点时的路径长度*。经过某点时的路径长度 = 左子树深度 + 右子树深度

代码

/**
 * 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:
    /*解题思路：最长的路径必然经过某个点，枚举以经过该点时的路径长度  cnblogs.com@toulanboy*/
    int getAns(TreeNode * root){//求以root为根节点的树的深度
        if(!root){//空节点在第0层
            return 0;
        }
        int left = getAns(root->left);//左子树深度
        int right = getAns(root->right);//右子树深度
        ans = max(ans, left+right);//经过该点时的路径长度 = 左 + 右
        return max(left, right)+1;//返回该树的深度
    }
    int diameterOfBinaryTree(TreeNode* root) {
        ans = 0;
        getAns(root);
        return ans;
    }
private:
    int ans;
};