【两次过】Lintcode 595. 二叉树最长连续序列

给一棵二叉树,找到最长连续路径的长度。
这条路径是指 任何的节点序列中的起始节点到树中的任一节点都必须遵循 父-子 联系。最长的连续路径必须是从父亲节点到孩子节点(不能逆序)。

样例

举个例子:

   1
    \
     3
    / \
   2   4
        \
         5

最长的连续路径为 3-4-5,所以返回 3

   2
    \
     3
    / 
   2    
  / 
 1

最长的连续路径为 2-3 ,而不是 3-2-1 ,所以返回 2


解题思路:

Traverse + Divide Conquer。用全局变量longest来存储最长长度。

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /**
     * @param root: the root of binary tree
     * @return: the length of the longest consecutive sequence path
     */
    public int longestConsecutive(TreeNode root) {
        // write your code here
        longest = 0;
        
        helper(root);
        
        return longest;
    }
    
    private int longest;
    
    //返回当前root最长连续路径长度
    private int helper(TreeNode root){
        if(root == null)
            return 0;
        
        //Divide
        int left = helper(root.left);
        int right = helper(root.right);
        
        int tempMax = 1;// at least we have root
        if(root.left != null && root.val+1 == root.left.val){
            tempMax = Math.max(tempMax, left+1);
        }
        
        if(root.right != null && root.val+1 == root.right.val){
            tempMax = Math.max(tempMax, right+1);
        }
        
        longest = Math.max(tempMax, longest);
        
        return tempMax;
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于一棵二叉树,我们可以定义其最长连续序列(LCS)为从根节点开始,沿着任意路径向下遍历,使得经过的结点值连续递增或递减的最长路径长度。 为了求解二叉树的 LCS,我们可以使用递归的思想。具体来说,我们可以定义一个递归函数 dfs,其参数为当前结点和上一个结点的值 pre,并返回以当前结点为结尾的 LIS 长度和 LDS 长度。对于一个结点 node,如果 node.val = pre + 1,则可以将当前结点加入到 LIS 中;如果 node.val = pre - 1,则可以将当前结点加入到 LDS 中。如果 node.val 既不是 pre + 1 也不是 pre - 1,则需要重新开始计算 LCS 长度。最终的 LCS 长度即为所有以各个结点为结尾的 LIS 和 LDS 长度的最大值。 下面是一个 Java 代码实现: ```java class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } } class Solution { public int longestConsecutive(TreeNode root) { if (root == null) { return 0; } int[] res = dfs(root, root.val - 1); return Math.max(res[0], res[1]); } private int[] dfs(TreeNode node, int pre) { if (node == null) { return new int[]{0, 0}; } int[] left = dfs(node.left, node.val); int[] right = dfs(node.right, node.val); int[] cur = new int[]{1, 1}; if (node.val == pre + 1) { cur[0] = Math.max(cur[0], left[0] + 1); cur[1] = Math.max(cur[1], right[1] + 1); } else if (node.val == pre - 1) { cur[0] = Math.max(cur[0], right[0] + 1); cur[1] = Math.max(cur[1], left[1] + 1); } return cur; } } ``` 其中 dfs 函数返回一个长度为 2 的数组,第一个元素表示以当前结点为结尾的 LIS 长度,第二个元素表示以当前结点为结尾的 LDS 长度。对于每个结点,我们分别计算其向左子树和右子树的 LIS 和 LDS 长度,并根据当前结点与上一个结点的大小关系来更新当前结点的 LIS 和 LDS 长度。最终返回根节点的 LIS 和 LDS 长度的最大值即可。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值