687. 最长同值路径

最长同值路径

题目描述:

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

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

示例 1:

输入:root = [5,4,5,1,1,5]
输出:2
示例 2:

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

提示:

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

思路:

时间复杂度:O(n),空间复杂度O()

dfs遍历看左边好还是右边好

代码:

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func longestUnivaluePath(root *TreeNode) int {

    var dfs func(root *TreeNode)int
    res := 0
    dfs = func(root *TreeNode)int{
        if root == nil{
            return 0
        }
        left1,right1 := 0,0
        left := dfs(root.Left)
        right := dfs(root.Right)
        if root.Left != nil && root.Val == root.Left.Val{
            left1 = left+1
        }
        if root.Right != nil && root.Val == root.Right.Val{
            right1 = right+1
        }
        res = max(res,left1+right1)
        return max(left1,right1)
    }
    dfs(root)
    return res
    
}

func max(a,b int)int{
    if a>b{
        return a
    }
    return b
}

代码效率:

执行用时:52 ms, 在所有 Go 提交中击败了94.78%的用户
内存消耗:7.3 MB, 在所有 Go 提交中击败了26.12%的用户


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!