Leetcode刷题（543. 二叉树的直径）

一.题目

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

示例:

给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

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

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

二.代码（C）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
int count(struct TreeNode* root,int *length,int leftlength,int rightlength);
int diameterOfBinaryTree(struct TreeNode* root)
{
    int a=0;
    int* length=&a,leftlength=0,rightlength=0;
    count(root,length,leftlength,rightlength);
    return *length;
}
int count(struct TreeNode* root,int *length,int leftlength,int rightlength)
{
    if(root)
    {
        if(root->left)
        {
           
            leftlength = count(root->left,length,leftlength,0)+1;
        }
        if(root->right)
        {
           
            rightlength = count(root->right,length,0,rightlength)+1;
        }
        if(*length<leftlength+rightlength)
        {
            *length = leftlength+rightlength;
        }
        
    } 
    if(leftlength<rightlength)
        return rightlength;
    return leftlength;
}

三.提交记录

四.备注

迭代，每次找到当前节点作为根的直径，对比找出最大值。