Fork me on GitHub

Leetcode687.Longest Univalue Path最长同值路径

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

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

示例 1:

输入:

5 / \ 4 5 / \ \ 1 1 5

输出:

2

示例 2:

输入:

1 / \ 4 5 / \ \ 4 4 5

输出:

2

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

 

 

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

class Solution {
public:
    int longestUnivaluePath(TreeNode* root) {
        if(root == NULL)
            return 0;
        int temp = max(Find1(root, root ->val), Find2(root, root ->val));
        if(temp >= 2)
            temp = temp - 1;
        else
            temp = 0;
        return max(temp, max(longestUnivaluePath(root ->left), longestUnivaluePath(root ->right)));
    }

    int Find1(TreeNode* root, int val)
    {
        if(root == NULL)
            return 0;
        if(root ->val == val)
            return 1 + Find2(root ->left, val) + Find2(root ->right, val);
        return 0;
    }

    int Find2(TreeNode * root, int val)
    {
        if(root == NULL)
            return 0;
        if(root ->val == val)
            return 1 + max(Find2(root ->left, val), Find2(root ->right, val));
        return 0;
    }
};

 

posted @ 2018-11-05 21:07  lMonster81  阅读(57)  评论(0编辑  收藏  举报
/*评论*/ /*top按钮*/

/* 网易云控件 */