问题描述:
给定一个二叉树的 root ,返回 最长的路径的长度 ,这个路径中的 每个节点具有相同值 。 这条路径可以经过也可以不经过根节点。两个节点之间的路径长度 由它们之间的边数表示。
如下图,输出为2。
这道题的难点在于使用递归,需要的结果却不是递归的返回值。
递归函数返回的是当前节点向任意一子树延伸的最长路径,以图中的根节点为例,它返回的是0,因为左右节点都不是1。若输入4,则返回1,因为向左或右的路径都是1,注意左右的路径虽然可以相连,但它不是递归函数要返回的值。
将左节点的函数返回值记为l,右节点记为r。另设l0和r0表示包含当前节点的路径长度,初始化为0,若当前节点与左节点值相同则l0=l+1,否则仍保持为0,右节点同理。
我们将需要的值存在全局变量res中。以最高的4节点为例,最长路径是左右相加,所以当l0+r0大于res时更新res。而递归函数则返回r0和l0中较大的那个。
public class Solution {
int res;
public int LongestUnivaluePath(TreeNode root) {
res=0;
DFS(root);
return res;
}
public int DFS(TreeNode root)
{
if(root==null)
return 0;
int l=DFS(root.left),r=DFS(root.right);
int l0=0,r0=0;
if(root.left!=null&&root.left.val==root.val)
l0=l+1;
if(root.right!=null&&root.right.val==root.val)
r0=r+1;
if(res<l0+r0)
res=l0+r0;
return Math.Max(l0,r0);
}
}