Flatten Binary Tree to Linked List

最新推荐文章于 2025-12-18 10:59:28 发布

原创最新推荐文章于 2025-12-18 10:59:28 发布 · 232 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode #tree #先序遍历

leetcode 专栏收录该内容

175 篇文章

订阅专栏

本文介绍将二叉树展平为链表的方法，包括非递归和两种递归方式。非递归方法通过循环实现，而递归方法则通过对节点进行前序遍历或将根节点的左右子树重新连接来完成转换。

题目描述：

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

The flattened tree should look like:

非递归方法如下：

public void flatten(TreeNode root){  
    while(root!=null){  
    	System.out.println("root:"+root.val);
        if(root.left!=null){ 
            TreeNode pre = root.left;  
            while(pre.right!=null)  
                pre = pre.right;  
            pre.right = root.right;  
            root.right = root.left;  
            root.left = null;  
        }  
        root = root.right;  
    }  
}

2种递归方法

第一个方法在rootnode上直接操作，先暂时保存左右子树，如果root有左子树，把root.right赋值为root.left,root.left赋值为null，再在已经是右子树的左子树上找rightmost的node，把之前存住的right子树放在rightmost.right上即可。做完一遍以后，这时root的左子树已经已经被嫁接到了右子树和root之间，原来的左子树变成null，可以进行root.right的递归了。

代码如下：

public class Solution {
    public void flatten(TreeNode root) {
        if(root == null) 
            return;
        if(root.left != null){
            TreeNode left = root.left;
            TreeNode right = root.right;
            root.left = null;
            root.right = left;
            TreeNode rightMost = root.right;
            while(rightMost.right != null){
                 rightMost = rightMost.right;
            }
            rightMost.right = right;
        }
        flatten(root.right);
    }
}

第2种方法对二叉树进行前序遍历，把每次找到的节点加到链表的前一个节点。
注意要用一个一元数组，这样才能保证里面的当前节点的上一个节点是先序遍历的前一个节点而不是父节点。

这个和recover binary search tree比较，那个题pre是全局变量，所以可以那样做！

代码如下：

public class Solution {
     public void flatten_re(TreeNode root, TreeNode[] current) {  
        if(root==null) {  
            return;  
        }  
        TreeNode left  = root.left;  
        TreeNode right = root.right;  
        if(left!=null) {  
            root.left = null;  
        }  
        current[0].right = root;  
        current[0] = root;  
        if(left!=null) {  
            flatten_re(left,current);  
        }  
        if(root.right!=null) {  
            flatten_re(right,current);  
        }  
        return;  
    }  
      
    public void flatten(TreeNode root) {  
        if(root==null) {  
            return;  
        }  
        if(root.left==null&&root.right==null) {  
            return;  
        }  
        TreeNode [] array = new TreeNode[1];  
        array[0] = root;  
        flatten_re(root,array);  
        return;  
    }  
}