Leetcode-687-最长同值路径

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

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

示例 1:

输入:


              5
             / \
            4   5
           / \   \
          1   1   5

输出:

2

示例 2:

输入:

              1
             / \
            4   5
           / \   \
          4   4   5

输出:

2

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


分析:

1.如果不仔细观察,很容易陷入到一个TreeNode误区,对于用java解答的小伙伴,在IDE中解这道题,很容导包 import javax.swing.tree.TreeNode;,结果发现,这个TreeNode是个interface呀,这怎么办?难道要找它的实现子类?一脸茫然。淡定,这个时候就需要考验审题能力了,在平台给出的答案区上方给出了注释状态的一段代码,不知道读者有没有留意到?

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

其实这个就是本题需要的TreeNode,当我们在平台上提交答案的时候,不需要定义这个class,平台验收答案时是带着这个实现的;而当我们在IDE上进行编码调试的时候,需要把这个TreeNode代码拷贝到IDE中,解开注释。然后,就可以正式开始解题了。

2.我们来分析一下题目的关键点。
二叉树:一个节点只有两个子节点,左和右。
路径:节点与节点之间的连线就是一条路径
同值路径:节点和与子节点或者父节点之间的值相同,那么这就是一条同值路径,本题求的就是这个最大值。
边数:可以理解为题目示例中连接节点的直线。

3.这一类问题,需要用递归的方法解决。在二叉树中,只能从根节点向下递归调用,可以通过root.left和root.right的方法得到两个子节点的TreeNode对象,由此一层一层递归调用下去,直到最底端的叶节点。

4.第一个可能遇到的坑:如果当前节点与左节点的值不同,那么左边的同值路径就会断掉,同理,如果和右节点值不同,右边的同值路径就会断掉。一旦断掉,那么当前节点的同值路径值就应该清零。

5.第二个可能遇到的坑:如果当前节点与子节点的值不同,那么从子节点连过来的同值路径就断掉了,而如果当前节点和父节点的值相同,需要注意,会从本节点向父节点连一个新的同值路径,与从子节点来的同值路径无关,不可累加。本题求解的是多条同值路径中最长的那条。

6.第三个可能遇到的坑:先看下图,这是平台在验证答案时给出的测试用例。


平台测试用例.png

看图,这个二叉树的最大同值路径就是由根节点26组成的路径,从根节点开始,能连起来的,值为26的节点一共是6个,它们之间的连线一共是5条,那么答案就是5吗?NONO,平台的期望值是4,明白了吗?这里有一个隐含条件,同值路径必须是单向的,就是用纸笔能一笔画完的,不能倒退着连。


根据这些分析,java解答如下:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int longestUnivaluePath(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int[] longestPath = new int[1];
        getUnivaluePathCount(root, longestPath);
        return longestPath[0];
    }

    private int getUnivaluePathCount(TreeNode node, int[] longestPath) {
        int left = 0;
        if (node.left != null) {
            left += getUnivaluePathCount(node.left, longestPath);
            if (node.val == node.left.val) {
                /* 如果本节点和左节点值相同,路径能继续,再累加1 */
                left += 1;
            } else {
                /* 否则,就说明左边的路径已经断开,本节点的左方向的单向路径值清零 */
                left = 0;
            }
        }

        int right = 0;
        if (node.right != null) {
            right += getUnivaluePathCount(node.right, longestPath);
            if (node.val == node.right.val) {
                /* 如果本节点和右节点值相同,路径能继续,再累加1 */
                right += 1;
            } else {
                /* 否则,就说明右边的路径已经断开,本节点的右方向的单向路径值清零 */
                right = 0;
            }
        }
        int currentLongestPath;
        if (node.left != null && node.right != null && node.left.val == node.right.val) {
            /* 如果左右节点的值相同,那么可以相加连接起来。但如果左右相同但和本节点值不同,那么left和right的值都是0 */
            currentLongestPath = left + right;
        } else {
            /* 否则,最大的路径值只能是左右路径的最大值 */
            currentLongestPath = Math.max(left, right);
        }
        if (currentLongestPath > longestPath[0]) {
            /* 最终返回的结果在这里,每次递归都是以当前节点为核心,计算最大路径值,在多次递归当中,仅保留最大的值 */
            longestPath[0] = currentLongestPath;
        }
        /* 递归返回的是本节点与左右两个子节点的同值路径的最大值,注意,如果本节点的值与左右节点的值都不同,那么返回的是0 */
        return Math.max(left, right);
    }
}

最后说一下,本答案中的 int[] longestPath 用的java引用传递的特性,这并不是唯一的方法,还可以定义一个成员变量 private int longestPath; 来记录所有的递归过程最大的同值路径,那样会让递归方法少一个参数,在大量的递归调用过程中速度会更快。希望读者们能学到这两种方法,在今后的开发中根据实际情况合理使用。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • B树的定义 一棵m阶的B树满足下列条件: 树中每个结点至多有m个孩子。 除根结点和叶子结点外,其它每个结点至少有m...
    文档随手记阅读 13,355评论 0 25
  • 第一章 绪论 什么是数据结构? 数据结构的定义:数据结构是相互之间存在一种或多种特定关系的数据元素的集合。 第二章...
    SeanCheney阅读 5,822评论 0 19
  • 因为之前就复习完数据结构了,所以为了保持记忆,整理了一份复习纲要,复习的时候可以看着纲要想具体内容。 树 树的基本...
    牛富贵儿阅读 7,012评论 3 10
  • 树 记录《剑指offer》中所有关于树的题目,以及LeetCode中的相似题目。 相关题目列表 题目 树是一种最常...
    wenmingxing阅读 1,448评论 2 13
  • 他们都老啦,我还年轻呢。有种瞌睡做了个长梦,其实只睡了一小会儿,醒来精力充沛时间还剩很多的赚到感。
    鼹吉吉阅读 233评论 2 1