这是悦乐书的第290次更新,第308篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第158题(顺位题号是687)。给定二叉树,找到路径中每个节点具有相同值的最长路径的长度。此路径可能会也可能不会通过根目录。例如:
输入:
5
/ \
4 5
/ \ \
1 1 5
输出:路径为[5,5,5],边长为2
输入:
1
/ \
4 5
/ \ \
4 4 5
输出:路径为[4,4,4],边长为2
注意:
两个节点之间的路径长度由它们之间的边数表示。
给定的二叉树不超过10000个节点。 树的高度不超过1000。
本次解题使用的开发工具是eclipse,jdk使用的版本是1.8,环境是win7 64位系统,使用Java语言编写和测试。
02 第一种解法
此题与求二叉树的最长路径边长相似,只是此题要求是节点值相同的路径,也就是说在找最长路径的时候,还需要判断节点值,要是不相同,就重置为0,在此期间,我们使用一个全局变量来存储最长节点值相同路径的边长。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int count = 0;
public int longestUnivaluePath(TreeNode root) {
helper(root);
return count;
}
public int helper(TreeNode node){
if (node == null) {
return 0;
}
int left = helper(node.left);
int right = helper(node.right);
if (node.left == null || node.left.val != node.val) {
left = 0;
}
if (node.right == null || node.right.val != node.val) {
right = 0;
}
count = Math.max(count, left+right);
return Math.max(left, right)+1;
}
}
03 第二种解法
如果你对上面的解法不太理解,尤其是递归方法最后一步返回left与right中的最大值再加1,可以看看此种解法。在处理完左子树的节点后,如果当前节点和其左子节点的节点值相等,那么left就在原基础上加1,否则就重置为0。因为要想最长,从当前节点开始,就是其左右子树中的最长相同节点值路径加上当前节点本身。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int count = 0;
public int longestUnivaluePath(TreeNode root) {
helper(root);
return count;
}
public int helper(TreeNode node){
if (node == null) {
return 0;
}
int left = helper(node.left);
int right = helper(node.right);
left = node.left != null && node.left.val == node.val ? left+1 : 0;
right = node.right != null && node.right.val == node.val ? right+1 : 0;
count = Math.max(count, left+right);
return Math.max(left, right);
}
}
04 小结
算法专题目前已日更超过四个月,算法题文章158+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。
以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!