[LeetCode] 114. Flatten Binary Tree to Linked List

原题链接：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，也就是说，需要在原树上进行更改，而不能另建新树。仔细观察可以发现，修改之后新树的顺序其实就是对原树的先序遍历顺序。

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

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

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

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

/**
 * 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;
        }
    }
}