乐趣区

关于java:LeetCode114二叉树展开为链表

二叉树开展为链表

题目形容:给你二叉树的根结点 root,请你将它开展为一个单链表:

  • 开展后的单链表应该同样应用 TreeNode,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null。
  • 开展后的单链表应该与二叉树 先序遍历 程序雷同。

示例阐明请见 LeetCode 官网。

起源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl…
著作权归领扣网络所有。商业转载请分割官网受权,非商业转载请注明出处。

解法一:递归

应用递归的形式求解,递归过程如下:

  • 如果以后根节点为 null 时,间接返回;
  • 如果以后根节点的左右子节点都为空时,不须要调整,间接返回;
  • 如果以后根节点的左子树为空,则只须要递归解决以后根节点的右子树;
  • 如果以后根节点的右子树为空时,则须要递归解决以后根节点的左子树,而后将以后根节点的左子节点指向 null,右子节点指向解决后的左子树;
  • 如果以后根节点的左右子树都不为空,则别离递归解决以后根节点的左右子树,而后将根节点的左子节点指向 null,右子节点指向解决后的左子树,而后将左子树的最初一个节点指向解决后的右子树。
import com.kaesar.leetcode.TreeNode;

public class LeetCode_114 {
    /**
     * 递归
     *
     * @param root
     */
    public static void flatten(TreeNode root) {
        // 以后根节点为 null 时,间接返回
        if (root == null) {return;}
        // 以后根节点的左右子节点都为空时,不须要调整,间接返回
        if (root.left == null && root.right == null) {return;}
        if (root.left == null) {
            // 以后根节点的左子树为空,则只须要递归解决以后根节点的右子树
            flatten(root.right);
            return;
        }
        if (root.right == null) {
            // 以后根节点的右子树为空时,则须要递归解决以后根节点的左子树,而后将以后根节点的左子节点指向 null,右子节点指向解决后的左子树
            TreeNode left = root.left;
            flatten(left);
            root.right = left;
            root.left = null;
            return;
        }
        // 如果以后根节点的左右子树都不为空
        TreeNode left = root.left;
        // 递归解决左子树,并找到解决后的最初一个节点
        flatten(left);
        TreeNode leftLast = left;
        while (leftLast.right != null) {leftLast = leftLast.right;}
        TreeNode right = root.right;
        // 递归解决右子树
        flatten(right);
        // 最初将根节点的左子节点指向 null,右子节点指向解决后的左子树,而后将左子树的最初一个节点指向解决后的右子树
        root.right = left;
        leftLast.right = right;
        root.left = null;
    }

    public static void main(String[] args) {TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(5);
        root.left.left = new TreeNode(3);
        root.left.right = new TreeNode(4);
        root.right.right = new TreeNode(6);

        System.out.println("开展为链表之前");
        root.print();
        System.out.println("开展为链表之后");
        flatten(root);
        root.print();}
}

【每日寄语】 咱们的人生像云霄飞车,充斥高下与刺激,咱们经验多,心理建设要强健,眼界开,襟怀要放大。

退出移动版