687. 最长同值路径 : 简单递归运用题-CFANZ编程社区

题目描述

这是 LeetCode 上的 687. 最长同值路径 ，难度为中等。

Tag : 「二叉树」、「DFS」、「递归」

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

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

示例 1:

687. 最长同值路径 : 简单递归运用题_递归

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

输出：2

示例 2:

687. 最长同值路径 : 简单递归运用题_递归_02

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

输出：2

提示:

树的节点数的范围是
树的深度将不超过1000

递归

设计递归函数 int dfs(TreeNode root)，含义为传入根节点 root，返回以该节点为起点，往下走同值路径所能经过的最大路径长度（即不能同时往左右节点走），同时使用全局变量 max 记录答案路径所能经过最大路径长度。

在递归函数内部，先通过递归 root 的左右子节点，拿到以 root.left 和 root.right 为起点的最大路径长度 l 和 r，然后根据当前节点值和左右子节点值的相等关系来更新 ans，同时用 cur 维护「以当前节点 root 为目标路径中深度最小（位置最高）节点时」所经过的最大路径长度。

Java 代码：

class Solution {
    int max = 0;
    public int longestUnivaluePath(TreeNode root) {
        dfs(root);
        return max;
    }
    int dfs(TreeNode root) {
        if (root == null) return 0;
        int ans = 0, cur = 0, l = dfs(root.left), r = dfs(root.right);
        if (root.left != null && root.left.val == root.val) {
            ans = l + 1; cur += l + 1;
        }
        if (root.right != null && root.right.val == root.val) {
            ans = Math.max(ans, r + 1); cur += r + 1;
        }
        max = Math.max(max, cur);
        return

TypeScript 代码：

let max = 0;
function longestUnivaluePath(root: TreeNode | null): number {
    max = 0
    dfs(root)
    return max
};
function dfs(root: TreeNode | null): number {
    if (root == null) return 0
    let ans = 0, cur = 0, l = dfs(root.left), r = dfs(root.right)
    if (root.left != null && root.left.val == root.val) {
        ans = l + 1; cur += l + 1
    }
    if (root.right != null && root.right.val == root.val) {
        ans = Math.max(ans, r + 1); cur += r + 1
    }
    max = Math.max(max, cur)
    return