【LeetCode】687. 最长同值路径

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/u012736685/article/details/88831781

一、题目描述

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

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

二、示例

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