二叉树的最长连续子序列 II - LintCode

描述
给定一棵二叉树,找到最长连续序列路径的长度。
路径起点跟终点可以为二叉树的任意节点。

样例

    1
   / \
  2   0
 /
3

返回 4 // 0-1-2-3

思路
对于每个节点root,求以root为起始节点的最长连续递增递增序列的长度up和最长连续递减序列的长度down,结果返回 < up,down >。初始化up, down均为1,left,right分别表示root左右子树的< up, down>,当root的左子树存在,若root左子树的值是root的值+1,up的值为左子树up+1;若root右子树的值是root的值-1,down的值为左子树down+1。当root的右子树存在时,也是同样的操作。

#ifndef C614_H
#define C614_H
#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
class TreeNode{
public:
    int val;
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 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、付费专栏及课程。

余额充值