【java】114. Flatten Binary Tree to Linked List

最新推荐文章于 2020-03-30 22:28:58 发布

原创最新推荐文章于 2020-03-30 22:28:58 发布 · 317 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode

leetcode 专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种将二叉树通过递归方式展平为链表的方法。该方法首先递归展平左右子树，然后重新连接节点，最终使每个节点的左子节点为空，右子节点指向原二叉树中的下一个节点。

问题原文https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/description/

方法：递归

public void flatten(TreeNode root) {
        if(root == null) return;
        TreeNode left = root.left;
        TreeNode right =root.right;
        flatten(left);
        flatten(right);
        root.left = null;
        if(left == null && right == null) return;
        if(left == null) {
        	root.right = right;
        	return;
        }
        if(right == null) {
        	root.right = left;
        	return;
        }
        TreeNode current = left;
        while(current.right!=null) {
        	current = current.right;
        }
        current.right = right;
        root.right = left;
    }