码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] 687. Longest Univalue Path

时间:2020-03-18 09:19:03      阅读:57      评论:0      收藏:0      [点我收藏+]

标签:get   example   例子   turn   inpu   class   span   tree   变量   

最长同值路径。题意是给一个二叉树,请输出一个最长的同值路径的长度。例子,

Example 1:

Input:

              5
             /             4   5
           / \             1   1   5

Output: 2

 

Example 2:

Input:

              1
             /             4   5
           / \             4   4   5

Output: 2

思路是后序遍历,可参考250题。同样也是需要创建一个全局变量记录这个路径长度。后序遍历递归的时候判断左孩子的val跟当前节点的val是否相等。res记录的是Math.max(res, left + right)。

时间O(n)

空间O(n)

Java实现

 1 class Solution {
 2     int res = 0;
 3 
 4     public int longestUnivaluePath(TreeNode root) {
 5         if (root == null)
 6             return 0;
 7         helper(root);
 8         return res;
 9     }
10 
11     private int helper(TreeNode root) {
12         if (root == null)
13             return 0;
14         int left = helper(root.left);
15         int right = helper(root.right);
16         if (root.left != null) {
17             left += root.left.val == root.val ? 1 : -left;
18         }
19         if (root.right != null) {
20             right += root.right.val == root.val ? 1 : -right;
21         }
22         res = Math.max(res, left + right);
23         return Math.max(left, right);
24     }
25 }

 

JavaScript实现

 1 /**
 2  * @param {TreeNode} root
 3  * @return {number}
 4  */
 5 var longestUnivaluePath = function (root) {
 6     var max = 0;
 7     var helper = function (root) {
 8         if (root == null) {
 9             return 0;
10         }
11         var left = helper(root.left);
12         var right = helper(root.right);
13         if (root.left) {
14             left += root.left.val === root.val ? 1 : -left;
15         }
16         if (root.right) {
17             right += root.right.val === root.val ? 1 : -right;
18         }
19         max = Math.max(max, left + right);
20         return Math.max(left, right);
21     };
22     helper(root);
23     return max;
24 };

 

[LeetCode] 687. Longest Univalue Path

标签:get   example   例子   turn   inpu   class   span   tree   变量   

原文地址:https://www.cnblogs.com/aaronliu1991/p/12515166.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!