687. 最长同值路径

687. 最长同值路径

1.题目描述

给定一个二叉树，找到最长的路径，这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意：两个节点之间的路径长度由它们之间的边数表示。
示例 1:
输入:

输出:

示例 2:
输入:

输出:

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

2.思路

深度优先遍历二叉树，如果当前节点的左子节点等于当前节点，则左路径leftpath = left + 1，如果当前节点的右子节点等于当前节点，则右路径rightpath = right + 1，遍历过程中取左右路径之和的最大值。

3.代码

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

4.复杂度分析

时间复杂度：O(n)
空间复杂度：O(n)