【码道初阶】【牛客BM30】二叉搜索树与双向链表:java中以引用代指针操作的艺术与陷阱

【牛客BM30】二叉搜索树与双向链表:java中以引用代指针操作的艺术与陷阱

在数据结构面试中,"将二叉搜索树(BST)转换成有序的双向链表" 是一道考察指针操作、递归思维以及边界条件处理的经典题目。

题目要求我们在 O(1)O(1)O(1) 空间复杂度 (原地操作)下完成转换,这意味着我们不能创建新节点,只能改变原有树节点 leftright 指针的指向。

今天我们就来拆解这个问题的核心思路,并复盘一个容易被忽视的"空指针陷阱"。

1. 核心思路:中序遍历 + 全局前驱

为什么是中序遍历?

二叉搜索树(BST)的一个核心性质是:其中序遍历(左 -> 根 -> 右)的结果是严格递增有序的

题目要求生成的双向链表也是有序的。因此,解题的大框架必然是中序遍历

怎么把树变成链表?

我们需要在遍历的过程中,修改节点的指针。

  • 树节点的 left 指针 →\rightarrow→ 双向链表的 prev 指针。
  • 树节点的 right 指针 →\rightarrow→ 双向链表的 next 指针。

关键变量:prev

为了将当前节点(cur)与前一个遍历到的节点连接起来,我们需要一个全局变量 prev 来记录**"在中序遍历中,上一个访问完的节点"**。

算法流程(非常类似中序遍历的递归写法)

只不过中序遍历是在递归左子树和右子树之间加上System.out.print("root.val" +" ");

而由图可以看出来,这二叉树转的双向链表顺序明显满足中序遍历。

  1. 递归左子树:先处理左边,把左边已经转换好。
  2. 处理当前节点(连接操作)
    • 将当前节点 root 的左指针指向 prev (root.left = prev)。
    • 如果 prev 不为空,将 prev 的右指针指向当前节点 (prev.right = root)。
    • 更新 prev :当前节点处理完毕,它变成了下一个节点的"前驱",所以 prev = root
  3. 递归右子树:继续处理右边。

2. 代码深度解析

java 复制代码
public class Solution {
    // 全局变量,记录中序遍历过程中"上一个"处理过的节点
    TreeNode prev = null;

    public TreeNode Convert(TreeNode pRootOfTree) {
        // 【问题核心】为什么要单独判断空?后面会详细解答
        if(pRootOfTree == null) return null;
        
        // 1. 执行中序遍历,修改指针
        ConvertChild(pRootOfTree);
        
        // 2. 寻找链表头节点
        // 转换完后,pRootOfTree 还在树的根节点位置(也就是链表的中间某处)
        // 双向链表的头节点应该是"最左侧"的节点
        TreeNode head = pRootOfTree;
        while(head.left != null){
            head = head.left;
        }
        
        return head;
    }

    // 辅助函数:标准的中序遍历框架
    public void ConvertChild(TreeNode root){
        if(root == null) return; // 递归终止条件
        
        // 1. 递归左子树
        ConvertChild(root.left);
        
        // 2. 核心连接逻辑
        root.left = prev;      // 当前节点的左指针指向前驱
        if(prev != null){
            prev.right = root; // 前驱的右指针指向当前节点(双向绑定)
        }
        prev = root;           // 移动 prev 指针,当前节点成为下一个节点的前驱
        
        // 3. 递归右子树
        ConvertChild(root.right);
    }
}

3. 灵魂拷问:为什么必须在 Convert 中加判空?

这是本题最容易踩坑的地方。

问题描述

明明在 ConvertChild 函数的第一行已经写了 if(root == null) return;,为什么在主函数 Convert 开头不加 if(pRootOfTree == null) return null; 就会导致部分测试用例(空树)不通过?

详细解答

这里的判空不是为了防止递归出错,而是为了防止后续寻找头节点时的空指针异常

让我们模拟一下输入为空树 {} 的情况:

  1. 假设没有 Convert 函数里的判空。
  2. 输入 pRootOfTreenull
  3. 调用 ConvertChild(null)。进入辅助函数,触发 if(root == null) return;,函数直接结束,没问题。
  4. 回到 Convert 主函数,继续往下执行。
  5. 执行 TreeNode head = pRootOfTree;,此时 head 被赋值为 null
  6. 执行 while(head.left != null)
    • 程序试图访问 null.left
    • 💥 崩!抛出 java.lang.NullPointerException

结论
ConvertChild 中的判空是递归的终止条件 (Base Case),它保证了递归能正常结束。

Convert 中的判空是防御性编程 ,它保护了后续寻找 head 的逻辑(head.left)不操作空对象。

如果不加这一句,当输入是空树时,程序会在寻找头节点时崩溃。

4. 寻找头节点的两种策略

在代码中,我们通过 while 循环往左走来寻找头节点。其实还有一种不需要循环的方法:

由于 prev 在遍历结束后会指向链表的尾节点 (中序遍历的最后一个节点),我们可以利用 prev 一路往左推(利用 left 指针),或者记录最开始的 head

但在本题的结构下,直接从 pRootOfTree 往左找 head 是最直观的,因为转换后的链表依然保持了 left 指向更小元素的特性。

5. 总结

这道题考察了三个核心点:

  1. 理解 BST 性质:中序遍历即有序。
  2. 双指针操作 :在遍历过程中动态修改 leftright,像缝衣服一样把节点串起来。
  3. 鲁棒性 :处理 input == null 的边界情况,避免后续逻辑空指针异常。
相关推荐
王老师青少年编程4 分钟前
信奥赛C++提高组csp-s之并查集(案例实践)2
数据结构·c++·并查集·csp·信奥赛·csp-s·提高组
范纹杉想快点毕业12 分钟前
嵌入式通信核心架构:从状态机、环形队列到多协议融合
linux·运维·c语言·算法·设计模式
东北小狐狸-Hellxz14 分钟前
解决java客户端连接ssh失败问题
java·网络·ssh
悟能不能悟15 分钟前
HttpServletRequest request获取整个headers有什么方法
java
__万波__16 分钟前
二十三种设计模式(二十)--解释器模式
java·设计模式·解释器模式
智源研究院官方账号26 分钟前
众智FlagOS 1.6发布,以统一架构推动AI硬件、软件技术生态创新发展
数据库·人工智能·算法·架构·编辑器·硬件工程·开源软件
网安_秋刀鱼27 分钟前
【java安全】反序列化 - CC1链
java·c语言·安全
零度@29 分钟前
Java消息中间件-Kafka全解(2026精简版)
java·kafka·c#·linq
钱多多_qdd31 分钟前
springboot注解(二)
java·spring boot·后端