【leetcode】687 最长同值路径（二叉树，递归）

题目链接：https://leetcode-cn.com/problems/longest-univalue-path/

题目描述

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

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

示例 1:

输入:

              5
             / \
            4   5
           / \   \
          1   1   5
输出:

2

示例 2:

输入:

              1
             / \
            4   5
           / \   \
          4   4   5
输出:

2

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

思路

与 124题 二叉树中的最大路径和思路相似
我们可以将任何路径（具有相同值的节点）看作是最多两个从其根延伸出的箭头。

具体地说，路径的根将是唯一节点，因此该节点的父节点不会出现在该路径中，而箭头将是根在该路径中只有一个子节点的路径。

然后，对于每个节点，我们想知道向左延伸的最长箭头和向右延伸的最长箭头是什么？我们可以用递归来解决这个问题。

令longestCore(root->left)为从节点root延伸出的最长箭头的长度；如果 root->left 存在且与节点 root 具有相同的值，则该值就会是 1 + longestCore(root->left)。在 root->right 存在的情况下也是一样。

代码

class Solution {
    int maxLen = 0;
public:
    int longestUnivaluePath(TreeNode* root) {
        if(!root) return 0;
        longestCore(root);
        return maxLen;
    }

    // 从root出发节点值为root->val的最长连续路径长度
    int longestCore(TreeNode* root){
        if(!root) return 0;
        int left = longestCore(root->left);
        int right = longestCore(root->right);
        int leftLen = 0, rightLen = 0;  // 左右部分值为root->val的连续路径长度
        if(root->left && root->left->val == root->val)
            leftLen = left + 1;         //
        if(root->right && root->right->val == root->val)
            rightLen = right + 1;
        maxLen = max(maxLen, leftLen+ rightLen);
        return max(leftLen, rightLen);
    }
};