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 panda库从基础到高级操作分析

《pythonpanda库从基础到高级操作分析》本文介绍了Pandas库的核心功能,包括处理结构化数据的Series和DataFrame数据结构,数据读取、清洗、分组聚合、合并、时间序列分析及大数据... 目录1. Pandas 概述2. 基本操作:数据读取与查看3. 索引操作:精准定位数据4. Group

Python pandas库自学超详细教程

《Pythonpandas库自学超详细教程》文章介绍了Pandas库的基本功能、安装方法及核心操作,涵盖数据导入(CSV/Excel等)、数据结构(Series、DataFrame)、数据清洗、转换... 目录一、什么是Pandas库(1)、Pandas 应用(2)、Pandas 功能(3)、数据结构二、安

Python使用Tenacity一行代码实现自动重试详解

《Python使用Tenacity一行代码实现自动重试详解》tenacity是一个专为Python设计的通用重试库,它的核心理念就是用简单、清晰的方式,为任何可能失败的操作添加重试能力,下面我们就来看... 目录一切始于一个简单的 API 调用Tenacity 入门:一行代码实现优雅重试精细控制:让重试按我

Python安装Pandas库的两种方法

《Python安装Pandas库的两种方法》本文介绍了三种安装PythonPandas库的方法,通过cmd命令行安装并解决版本冲突,手动下载whl文件安装,更换国内镜像源加速下载,最后建议用pipli... 目录方法一:cmd命令行执行pip install pandas方法二:找到pandas下载库,然后

Python实现网格交易策略的过程

《Python实现网格交易策略的过程》本文讲解Python网格交易策略,利用ccxt获取加密货币数据及backtrader回测,通过设定网格节点,低买高卖获利,适合震荡行情,下面跟我一起看看我们的第一... 网格交易是一种经典的量化交易策略,其核心思想是在价格上下预设多个“网格”,当价格触发特定网格时执行买

Python标准库之数据压缩和存档的应用详解

《Python标准库之数据压缩和存档的应用详解》在数据处理与存储领域,压缩和存档是提升效率的关键技术,Python标准库提供了一套完整的工具链,下面小编就来和大家简单介绍一下吧... 目录一、核心模块架构与设计哲学二、关键模块深度解析1.tarfile:专业级归档工具2.zipfile:跨平台归档首选3.

使用Python构建智能BAT文件生成器的完美解决方案

《使用Python构建智能BAT文件生成器的完美解决方案》这篇文章主要为大家详细介绍了如何使用wxPython构建一个智能的BAT文件生成器,它不仅能够为Python脚本生成启动脚本,还提供了完整的文... 目录引言运行效果图项目背景与需求分析核心需求技术选型核心功能实现1. 数据库设计2. 界面布局设计3

Python进行JSON和Excel文件转换处理指南

《Python进行JSON和Excel文件转换处理指南》在数据交换与系统集成中,JSON与Excel是两种极为常见的数据格式,本文将介绍如何使用Python实现将JSON转换为格式化的Excel文件,... 目录将 jsON 导入为格式化 Excel将 Excel 导出为结构化 JSON处理嵌套 JSON:

Python操作PDF文档的主流库使用指南

《Python操作PDF文档的主流库使用指南》PDF因其跨平台、格式固定的特性成为文档交换的标准,然而,由于其复杂的内部结构,程序化操作PDF一直是个挑战,本文主要为大家整理了Python操作PD... 目录一、 基础操作1.PyPDF2 (及其继任者 pypdf)2.PyMuPDF / fitz3.Fre

python设置环境变量路径实现过程

《python设置环境变量路径实现过程》本文介绍设置Python路径的多种方法:临时设置(Windows用`set`,Linux/macOS用`export`)、永久设置(系统属性或shell配置文件... 目录设置python路径的方法临时设置环境变量(适用于当前会话)永久设置环境变量(Windows系统