Java 实现二叉树展平为链表

最新推荐文章于 2025-02-03 00:39:36 发布

布说在见

最新推荐文章于 2025-02-03 00:39:36 发布

阅读量1.2k

点赞数 17

CC 4.0 BY-SA版权

文章标签： java 链表开发语言

本文链接：https://blog.csdn.net/m0_67187271/article/details/141810983

Java 实现二叉树展平为链表

前言
使用原地算法（O(1) 空间复杂度）将二叉树展平为链表

前言

处理二叉树节点，迭代连接右子节点，移至左侧并清除左子连接，整合进链表，右子节点待处理

在处理二叉树数据结构时，有时需要将其转换成一种特殊的形态，即链表。

这种转换可以简化某些算法的操作，例如遍历或访问树中的节点。

本文将介绍如何使用Java编程语言将一个二叉树展平为链表，使得每个节点仅具有一个右子节点。

问题背景

给定一个二叉树，要求将该二叉树转换为一个链表，其中每个节点都只拥有一个右子节点。转换后的链表应该按照原二叉树的前序遍历顺序排列节点。

解决方案

为了达成目标，我们可以通过反复迭代的方法来逐个处理二叉树的节点。

在每次迭代中，若当前节点拥有左子树，我们便寻找该左子树的最右侧节点，并将其右子节点连接到当前节点的右侧。

随后，将当前节点的原始右子节点移至左侧，并清除其左子节点连接。

通过这种方式，原节点被整合进链表结构中，同时其右侧子节点成为下一个待处理的对象。

代码实现

首先定义二叉树节点类 TreeNode：

public class TreeNode {
   
   
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
   
   
        val = x;
    }
}

然后定义解决方案类 Solution，并在其中实现 flatten 方法：

public class Solution {
   
   
    public void flatten(TreeNode root) {
   
   
        while (root != null) {
   
   
            if (root.left == null) {
   
   
                // 如果当前节点没有左子树，则移动到右子树
                root = root.right