0x3f 第36天 外卖8,9,树

1.二叉搜索树中第k小的元素

1.核心思路:中序

2.中序出来的就是123456

3.所以每次左子树遍历之后,k-1 当k==0的时候,正好就是顺序的第k个

4.终止条件是遍历到 None或者k==0也就是提前找到了第k个,就直接结束,不用再找了

❌ 情况 1:不加 k==0 这个终止条件,代码会怎么样?

如果终止条件只写 if node is None: return,代码逻辑上能得到正确答案,但会做大量「无用功」:

  • 哪怕找到了目标节点(k=0),递归依然会继续执行:遍历完当前节点的右子树、遍历完父节点的右子树、遍历完整棵树的所有剩余节点;

  • 比如找第 2 小的节点,找到后还要遍历后面所有节点,时间复杂度从最优的 O(H+k) 变成了最差的 O(N)(N 是总节点数),效率暴跌!

    class Solution:
    def kthSmallest(self, root: Optional[TreeNode], k: int) -> int:
    ans = 0
    def dfs(node):
    nonlocal ans,k
    if node is None or k==0:
    return
    dfs(node.left)
    k-=1
    if k==0:
    ans = node.val
    dfs(node.right)
    dfs(root)
    return ans

2.二叉树展开为链表

  • 展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null

因为要求展开后的单链表应该与二叉树 先序遍历 顺序相同

那使用头插法就得用后序遍历

定义一个全局变量头head,根据要求,后序的root的right指向head,head再指向root,就完成了头插法,不好理解的话,画个图一下就明白了

相关推荐
robch14 小时前
golang container/heap 是一个为任意类型实现堆(优先队列)接口的包
数据结构·算法·golang
967714 小时前
力扣面试经典150 88. 合并两个有序数组 归并排序的merge函数
算法·leetcode·面试
菜鸡儿齐16 小时前
Unsafe方法学习
java·python·学习
老师好,我是刘同学20 小时前
Python执行命令并保存输出到文件
python
啵啵鱼爱吃小猫咪1 天前
机械臂阻抗控制github项目-mujoco仿真
开发语言·人工智能·python·机器人
MaximusCoder1 天前
等保测评命令——Centos Linux
linux·运维·经验分享·python·安全·centos
yunyun321231 天前
用Python生成艺术:分形与算法绘图
jvm·数据库·python
放下华子我只抽RuiKe51 天前
算法的试金石:模型训练、评估与调优的艺术
人工智能·深度学习·算法·机器学习·自然语言处理·数据挖掘·线性回归
m0_662577971 天前
高级爬虫技巧:处理JavaScript渲染(Selenium)
jvm·数据库·python
oem1101 天前
C++中的享元模式实战
开发语言·c++·算法