我的解法:
当时没有想到好的方法,潜意识里把递归排除了(以为会超时)
实际发现是可以用递归做的
需要注意序列需要单调增或者单调减,像[1,2,1]这样的序列虽然只差1也不符合需求
他人解法:
用多个变量统计,递归的时候返回两个信息,“底层到当前节点最长的递增连续路径长度“以及“底层到当前节点最长的递减连续路径长度”
因为是从底层开始递归,所以如果是子节点比父节点大,从子节点到父节点看上去就是递减。
class Solution:
def __init__(self):
self.ans = 1
def longestConsecutive(self, root: Optional[TreeNode]) -> int:
def dfs(root: TreeNode) -> tuple[int]:
if not root:
return 0
left = dfs(root.left)
right = dfs(root.right)
leftDec, rightInc, leftInc, rightDec = 0, 0, 0, 0
if root.left:
if root.left.val == root.val + 1:
leftDec = 1 + left[1] # 接上左边的最长递减长度
if root.left.val == root.val - 1:
leftInc = 1 + left[0] # 接上左边的最长递增长度
if root.right:
if root.right.val == root.val - 1:
rightInc = 1 + right[0]
if root.right.val == root.val + 1:
rightDec = 1 + right[1]
self.ans = max(self.ans, 1 + leftDec + rightInc, 1 + leftInc + rightDec)
return (max(leftInc, rightInc), max(leftDec, rightDec))
dfs(root)
return self.ans