[算法] 最长的单一路径

码农升级记

已于 2022-06-13 23:30:21 修改

阅读量199

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签：算法 leetcode 链表

于 2022-06-13 23:05:12 首次发布

原文链接：https://www.lintcode.com/problem/1085/solution/58212

算法专栏收录该内容

11 篇文章

订阅专栏

该博客介绍了一种解决算法问题的方法，即在给定的二叉树中寻找最长路径，路径上的所有节点值都相同。这个问题通过递归算法来解决，代码示例展示了如何遍历树并计算最长路径的长度。递归过程中，计算左右子树的最长路径，并更新当前节点的最长路径。最终返回的结果是整棵树中的最长相同值路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

说明：算法题目来源于lintcode，仅限于本人学习使用。请支持正版！

题目

给定一颗二叉树，找到路径中每个节点具有相同值的最长路径的长度。此路径可能会也可能不会通过根节点。

解题代码

使用递归算法。
本次代码属于转载。原代码请参考：https://www.lintcode.com/problem/1085/solution/58212

    /**
     * 思路解析：递归调用
     *       4
     *      / \
     *     4   5
     *    /   / \
     *   4   5   5
     *  /   /
     * 5   5
     * 上面的最长路径应该为四个5连接3条边
     * 递归思想。
     * 在 int left = arrowLength(node.left); 这步时，会递归算出左边的最大长度，最第四层的节点开始倒推，到第一层节点，长度为2
     * 在 int right = arrowLength(node.right); 这步时，会递归算出右边的最大长度。
     * 
     * 第三层：左分支5与三层节点5相等。arrowLeft = 1
     * 第二层：左分支5与第二层节点5相等。 arrowLeft = 1+1; arrowRight = 1 ; ans = 2+1
     * 第一层：左分支4与第二层4相等。arrowLeft = 1+1； arrowRight = 2，所以最大值为 ans = 2+1 = 3
     * @param root
     * @return
     */
    public int longestUnivaluePath(TreeNode root) {
        // Write your code here
        ans = 0;
        arrowLength(root);
        return ans;
    }
    public int arrowLength(TreeNode node) {
        if (node == null) return 0;
        int left = arrowLength(node.left);
        int right = arrowLength(node.right);
        int arrowLeft = 0, arrowRight = 0;
        if (node.left != null && node.left.val == node.val) {
            arrowLeft += left + 1;
        }
        if (node.right != null && node.right.val == node.val) {
            arrowRight += right + 1;
        }
        ans = Math.max(ans, arrowLeft + arrowRight);
        return Math.max(arrowLeft, arrowRight);
    }