leetcode_力扣_687. 最长同值路径

题目描述

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

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

示例

• 示例 1：
  

输入：root = [5,4,5,1,1,5]
输出：2

• 示例 2：
  

输入：root = [1,4,5,4,4,5]
输出：2

提示

• 树的深度将不超过 1000
• -1000 <= Node.val <= 1000
• 树的节点数的范围是 [0, 104]

解题思路

• 深度优先算法解题，注意父节点与子节点值不同时，左右路径长度的更新
• 设置全局变量，记录路径最大值

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int temp;
    
    int my_func(TreeNode *root){

        if(root == NULL){
            return 0;
        }

        int left, right, left_, right_;

        left = my_func(root->left);
        if(root->left != NULL && root->val == root->left->val){
            left_ = left + 1;
        }else{
            left_ = 0;
        }
        right = my_func(root->right);
        if(root->right != NULL && root->val == root->right->val){
            right_ = right + 1;
        }else{
            right_ = 0;
        }

        temp = temp > (left_ + right_) ? temp : (left_ + right_);
        
        return left_ > right_ ? left_ : right_;
    }

    int longestUnivaluePath(TreeNode* root) {

        temp = 0;
        my_func(root);

        return temp;
    }
};

提交结果

总结

• 深度优先遍历