LeetCode-114. Flatten Binary Tree to Linked List

题目描述

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

For example,
Given

     1    / \   2   5  / \   \ 3   4   6

The flattened tree should look like:
1
\
2
\
3
\
4
\
5
\
6

解题思路

采用递归的方式求解，对于每一个节点root，保存该节点的右节点到临时变量，对左右节点分别递归调用。先处理左节点，并将处理的结果作为该root节点的右节点（这就是前面为什么保存右节点的原因），然后找到当root节点的右节点的尾部tmp,处理右节点，并将结果作为tmp的右节点，最后返回root节点。

代码

/** * 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) {        if(root == null)            return;        root = flatternCore(root);        // Queue<TreeNode> queue = new LinkedList<TreeNode>();        // while()    }    private TreeNode flatternCore(TreeNode root){        TreeNode right = root.right;        root.right = null;//必须清空右节点，如果左节点不为空会将右节点覆盖，但是如果左节点为空，则会在计算root的右节点时出问题。        if(root.left != null){            root.right = flatternCore(root.left);            root.left = null;        }        if(right != null){            TreeNode tmp = root;            while(tmp.right != null){                tmp=tmp.right;            }            tmp.right=flatternCore(right);        }        return root;    }}