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,就完成了头插法,不好理解的话,画个图一下就明白了

相关推荐
测试19984 小时前
软件测试 - 单元测试总结
自动化测试·软件测试·python·测试工具·职场和发展·单元测试·测试用例
曲幽6 小时前
我用了FastApiAdmin后,连夜把踩过的坑都整理出来了
redis·python·postgresql·vue3·fastapi·web·sqlalchemy·admin·fastapiadmin
心中有国也有家7 小时前
cann-recipes-infer:昇腾 NPU 推理的“菜谱集合”
经验分享·笔记·学习·算法
前端若水8 小时前
会话管理:创建、切换、删除对话历史
前端·人工智能·python·react.js
绝知此事8 小时前
【算法突围 01】线性结构与哈希表:后端开发的收纳术
java·数据结构·算法·面试·jdk·散列表
碧海银沙音频科技研究院8 小时前
通话AEC与语音识别AEC的软硬回采链路
深度学习·算法·语音识别
涛声依旧-底层原理研究所8 小时前
残差连接与层归一化通俗易懂的详解
人工智能·python·神经网络·transformer
csdn_aspnet8 小时前
Python 算法快闪 LeetCode 编号 70 - 爬楼梯
python·算法·leetcode·职场和发展
fantasy_arch9 小时前
pytorch人脸匹配模型
人工智能·pytorch·python