687. 最长同值路径

687.最长同值路径

给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。

**注意:**两个节点之间的路径长度由它们之间的边数表示。

示例 1:

输入:

          5
         / \
        4   5
       / \   \
      1   1   5

输出:

2
示例 2:

输入:

          1
         / \
        4   5
       / \   \
      4   4   5

输出:

2
注意: 给定的二叉树不超过10000个结点。 树的高度不超过1000。

分析: 每个子树的最长同值路径不是其左子树的最长同值路径,就是右子树的最长同值路径就是两者之和。递归求解。对于某个节点,判断其左孩子的值是否与其相等,相等则;left等于1+左子树的最长同值路径,否则直接处理左子树,对右子树做相同计算。计算left、right之和,更新最大值,返回left和right的最大值。

/**
 * 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 longestUnivaluePath(TreeNode* root) {
        if(root==NULL) return 0;
        int max=0;
        getpath(root,max);
        return max;
    }
    
    int getpath(TreeNode* root,int &max)
    {
        int left=0;
        int right=0;
        if(root->left!=NULL&&root->val==root->left->val)
            left=1+getpath(root->left,max);
        else if(root->left!=NULL)
            getpath(root->left,max);
        if(root->right!=NULL&&root->val==root->right->val)
            right=1+getpath(root->right,max);
        else if(root->right!=NULL)
            getpath(root->right,max);
        if(left+right>max) max=left+right;
        return left>right?left:right;
        
    }
};

你可能感兴趣的:(LeetCode--C++)