LeetCode687----最长同值路径

Posted 2020-11-19 book808

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

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

示例 1:

输入:

              5
             / \\
            4   5
           / \\   \\
          1   1   5

输出:

2

示例 2:

输入:

              1
             / \\
            4   5
           / \\   \\
          4   4   5

输出:

2

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

 

思路：我们可以通过树结构发现，我们的每一个不同的同值路径（例如同值为2，或者同值为3），他们之间都是互不影响的，每次从一个同值路径跳到另一个同值路径时，前一个同值路径的状态不会影响到后一个同值路径：

 

当同值为4的路径走到3时，这时就应该断开4的状态，一切从0开始，直到同值为5的路径开始，于是乎上图我们又可以这样来画！

 

既然状态会断开，那么这里的3我们可以默认省去，这样，程序不论是走到最左下角的4节点还是最左下角的5节点，默认返回给其父节点的都是0

代码如下：

class Solution {
    public int max;
    
    public int longestUnivaluePath(TreeNode root) {
		help(root);
		return max;
	}

	public int help(TreeNode root) {
		if (root == null) {
			return 0;
		}
		int l = help(root.left);// 得到左边最大的路径
		int r = help(root.right);// 得到右边最大的路径
        int left = 0,right = 0;
		if (root.left != null && root.val == root.left.val) {
			// 证明根节点和当前的左子节点在一个路径
			left=l + 1;
		}
		if (root.right != null && root.val == root.right.val) {
			right=r + 1;
		}
		max = Math.max(max, left + right);
		return Math.max(left, right);
	}
}