リコウの質問：687。同じ値を持つ最長経路

リコウの質問：687。同じ値を持つ最長経路

バイナリツリーを指定して、最長のパスを見つけます。このパスの各ノードは同じ値を持っています。このパスは、ルートノードを通過する場合と通過しない場合があります。

注：2つのノード間のパスの長さは、ノード間のエッジの数で表されます。

例1：

入る：

          5
         / \
        4   5
       / \   \
      1   1   5

出力：

2
例2：

入る：

          1
         / \
        4   5
       / \   \
      4   4   5

出力：

2

ソース：LeetCode
リンク：https ://leetcode-cn.com/problems/longest-univalue-path
著作権はLeetCode が所有しています。商用転載については、正式な許可書にご連絡ください。非商用転載については、出典を明記してください。

コメント領域@Jingfuのソリューションを参照してください：Java公式ソリューションの思考分析：https://github.com/echofoo/ARTS/blob/master/A-20190925-longest path with the同じvalue.md

/**
 * 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 ret;
    int get_maxlen(TreeNode* root){
    
    
        if (root == NULL){
    
    
            return 0;//叶子节点，返回0
        }
        int leftlen=get_maxlen(root->left);
        int rightlen=get_maxlen(root->right);
        int left=0,right=0; //当左右子树不相等的时候，不会进入if条件，将返回0
        if (root->right && root->val==root->right->val){
    
    
            right = rightlen+1;
        }
        if (root->left && root->val==root->left->val){
    
    
            left=leftlen+1;
        }
        ret = ret>left+right?ret:left+right;//如果递归回到根节点，那么这个就是我们最终需要的值
        return left>right?left:right;//返回左右子树中长度较长的一个分支
    }
    int longestUnivaluePath(TreeNode* root) {
    
    
        ret = 0;
        get_maxlen(root);
        return ret;
    }
};