LeetCode面试298,二叉树最长连续序列(Python)

2024-05-04 08:20

本文主要是介绍LeetCode面试298,二叉树最长连续序列(Python),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

开始想着dfs,两种情况

1.以root为根

2.不以root为根

但是这样需要两个dfs分别进行,那么时间复杂度就上去了。

class Solution:def longestConsecutive(self, root: Optional[TreeNode]) -> int:def dfs(root):# 以root为根节点,可以延续几个点ans = 1if root.left and root.left.val - root.val == 1:ans = max(ans, 1+dfs(root.left))if root.right and root.right.val - root.val == 1:ans = max(ans, 1+dfs(root.right))return ansif not root: return 0ans = dfs(root)ans = max(ans, self.longestConsecutive(root.left), self.longestConsecutive(root.right))return ans

题解中提出一种思路可以同时进行dfs,只用把所有节点遍历一遍。

从上到下遍历,建立一个dfs(u, v, length),u为v的父节点,v为u的子节点,length以父节点为最后一个节点的序列长度(初始长度为1)

如果子节点刚好比父节点大1,那么length + 1,反之,length = 1

再继续遍历v的子节点

class Solution:def __init__(self):self.m = 1def longestConsecutive(self, root: Optional[TreeNode]) -> int:self.dfs(root, root, 1)return self.mdef dfs(self, u, v, length):# u为父节点,v为子节点,length为初始长度if not v: return if v.val - u.val == 1:length += 1self.m = max(self.m, length)else:length = 1self.dfs(v, v.left, length)self.dfs(v, v.right, length)

举一反三,还是按照之前的想法,不过只需要改变一点点,只有一种情况:

以root为根的最长序列

如果root.left.val - root.val==1,那么(以root为根的最长序列)可能就等于(以root.left为根的最长序列+1),如果root.right.val - root.val==1,那么(以root为根的最长序列)可能就等于(以root.right为根的最长序列+1),如果都不满足,那么返回1。

怎么有点像列表维护最长序列。其实就可以将链表想象成多方向的列表。

class Solution:def __init__(self):self.m = 1def longestConsecutive(self, root: Optional[TreeNode]) -> int:self.dfs(root)return self.mdef dfs(self, root):if not root:returnl, r = self.dfs(root.left), self.dfs(root.right)if root.left and root.left.val - root.val == 1:l += 1self.m = max(self.m, l)else:l = 0if root.right and root.right.val - root.val == 1:r += 1self.m = max(self.m, r)else:r = 0return max(1, l, r)

这篇关于LeetCode面试298,二叉树最长连续序列(Python)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/958752

相关文章

Python的Darts库实现时间序列预测

《Python的Darts库实现时间序列预测》Darts一个集统计、机器学习与深度学习模型于一体的Python时间序列预测库,本文主要介绍了Python的Darts库实现时间序列预测,感兴趣的可以了解... 目录目录一、什么是 Darts?二、安装与基本配置安装 Darts导入基础模块三、时间序列数据结构与

Python正则表达式匹配和替换的操作指南

《Python正则表达式匹配和替换的操作指南》正则表达式是处理文本的强大工具,Python通过re模块提供了完整的正则表达式功能,本文将通过代码示例详细介绍Python中的正则匹配和替换操作,需要的朋... 目录基础语法导入re模块基本元字符常用匹配方法1. re.match() - 从字符串开头匹配2.

Python使用FastAPI实现大文件分片上传与断点续传功能

《Python使用FastAPI实现大文件分片上传与断点续传功能》大文件直传常遇到超时、网络抖动失败、失败后只能重传的问题,分片上传+断点续传可以把大文件拆成若干小块逐个上传,并在中断后从已完成分片继... 目录一、接口设计二、服务端实现(FastAPI)2.1 运行环境2.2 目录结构建议2.3 serv

通过Docker容器部署Python环境的全流程

《通过Docker容器部署Python环境的全流程》在现代化开发流程中,Docker因其轻量化、环境隔离和跨平台一致性的特性,已成为部署Python应用的标准工具,本文将详细演示如何通过Docker容... 目录引言一、docker与python的协同优势二、核心步骤详解三、进阶配置技巧四、生产环境最佳实践

Python一次性将指定版本所有包上传PyPI镜像解决方案

《Python一次性将指定版本所有包上传PyPI镜像解决方案》本文主要介绍了一个安全、完整、可离线部署的解决方案,用于一次性准备指定Python版本的所有包,然后导出到内网环境,感兴趣的小伙伴可以跟随... 目录为什么需要这个方案完整解决方案1. 项目目录结构2. 创建智能下载脚本3. 创建包清单生成脚本4

Python实现Excel批量样式修改器(附完整代码)

《Python实现Excel批量样式修改器(附完整代码)》这篇文章主要为大家详细介绍了如何使用Python实现一个Excel批量样式修改器,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一... 目录前言功能特性核心功能界面特性系统要求安装说明使用指南基本操作流程高级功能技术实现核心技术栈关键函

python获取指定名字的程序的文件路径的两种方法

《python获取指定名字的程序的文件路径的两种方法》本文主要介绍了python获取指定名字的程序的文件路径的两种方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要... 最近在做项目,需要用到给定一个程序名字就可以自动获取到这个程序在Windows系统下的绝对路径,以下

使用Python批量将.ncm格式的音频文件转换为.mp3格式的实战详解

《使用Python批量将.ncm格式的音频文件转换为.mp3格式的实战详解》本文详细介绍了如何使用Python通过ncmdump工具批量将.ncm音频转换为.mp3的步骤,包括安装、配置ffmpeg环... 目录1. 前言2. 安装 ncmdump3. 实现 .ncm 转 .mp34. 执行过程5. 执行结

Python实现批量CSV转Excel的高性能处理方案

《Python实现批量CSV转Excel的高性能处理方案》在日常办公中,我们经常需要将CSV格式的数据转换为Excel文件,本文将介绍一个基于Python的高性能解决方案,感兴趣的小伙伴可以跟随小编一... 目录一、场景需求二、技术方案三、核心代码四、批量处理方案五、性能优化六、使用示例完整代码七、小结一、

Python中 try / except / else / finally 异常处理方法详解

《Python中try/except/else/finally异常处理方法详解》:本文主要介绍Python中try/except/else/finally异常处理方法的相关资料,涵... 目录1. 基本结构2. 各部分的作用tryexceptelsefinally3. 执行流程总结4. 常见用法(1)多个e