LeetCode 114 Flatten Binary Tree to Linked List

来源:互联网 发布:sql server培训机构 编辑:程序博客网 时间:2024/05/13 02:49
题目


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


思路


1 能够想到这其实是中序遍历的变形,就成功一半了。
2 一个节点左边的遍历好,然后处理一下,再遍历右边。
3处理的部分为:这个节点的左边设为null,右边接左边处理好的内容,再接原右边。
4 逻辑清楚了,代码自然也就清楚。

代码


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


15.4.14
可以用递归算法,更加简洁。看代码就能够理解意思
public class Solution {    public void flatten(TreeNode root) {        if(root == null){            return;        }        while(root!=null){            if(root.left!=null){                TreeNode cur = root.left;                while(cur.right!=null){                    cur = cur.right;                }                cur.right = root.right;                root.right = root.left;                root.left =null;            }            root = root.right;        }    }}



0 0
原创粉丝点击