Flatten Binary Tree to Linked List

最新推荐文章于 2020-06-07 20:07:19 发布

原创最新推荐文章于 2020-06-07 20:07:19 发布 · 3.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leecode #树操作

LeetCode 专栏收录该内容

85 篇文章

订阅专栏

本文介绍了一种将二叉树结构通过特定算法转换为链式结构的方法。主要思路是将每个节点的左子树连接到其右子树，并确保左子树成为新的右子树的一部分，同时保留原始右子树的链接。此过程递归进行，直至所有节点按顺序连接。

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

For example,
Given

The flattened tree should look like:

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

题目很简单，就是将一棵树化成一棵右单支树，过程是：将节点的作节点接到右节点的位置，原来的右节点接到现在右节点（也就是左节点）的最右节点的右子树，重复这个过程，直到是一棵右单支树。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public void flatten(TreeNode root) {
        if(root == null){
            return;
        }
        TreeNode p = root;
        TreeNode left = null;
        TreeNode right = null;
        TreeNode q = null;
        while(p != null ){//空节点退出
             left = p.left;
             right = p.right;
             if(left != null){//左子树不空，需要知道左子树的最右节点
                 q = ButtomRight(left);
                 p.left = null;
                 p.right = left;
                 q.right = right;
                 p = p.right;
             }else{//左子树为空
                 p = p.right; 
             }
        }
    }
    public TreeNode ButtomRight(TreeNode root){//返回root的最右节点
        TreeNode tmp = root;
        while(root.right != null){
            tmp = root.right;
            root = root.right;
        }
        return tmp;
    }
}