[LeetCode] 114. Flatten Binary Tree to Linked List

最新推荐文章于 2022-09-06 16:04:14 发布

原创最新推荐文章于 2022-09-06 16:04:14 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #树

LeetCode 同时被 2 个专栏收录

165 篇文章

订阅专栏

树

45 篇文章

订阅专栏

本文详细解析了LeetCode上的一道经典题目：如何将二叉树原地转化为只有右节点的链表，并遵循先序遍历的顺序。通过递归和指针操作，文章提供了一种有效的解决方案。

原题链接：https://leetcode.com/problems/flatten-binary-tree-to-linked-list/

1. 题目介绍

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

For example, given the following tree:
给出一个二叉树，将其转化为一棵只有右节点的树，并且需要按照先序遍历的顺序排列。
比如，给出下面的树：

在这里插入图片描述
The flattened tree should look like:
修改之后的树为：

2. 解题思路

本题给出的函数返回值为void，也就是说，需要在原树上进行更改，而不能另建新树。仔细观察可以发现，修改之后新树的顺序其实就是对原树的先序遍历顺序。

于是，问题转化成为，如何把原树改成先序遍历的顺序？

对于每个根节点，我们可以使用如下策略修改：

寻找自己的左子节点，假设它为节点A
在左子节点处，一直寻找右子节点，直到最后一个右子节点，假设它为节点B
将原来根节点的右子树，拼接到节点B的右子节点
将节点A改为根节点的右子节点
将根节点的左子树置空

具体实现代码如下所示：
实现代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public void flatten(TreeNode root) {
        while(root != null){
            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;
        }
    }
}