力扣114_二叉树展开为链表

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

展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。
在这里插入图片描述
算法思想:
这道题的函数是 void flatten(TreeNode root) ,我们尝试给出函数定义:给 flatten 函数输入一个节点 root,那么以 root 为根的二叉树就会被拉平为一条链表。
具体流程如下:

  1. 将 root 的左子树和右子树拉平。
  2. 将 root 的右子树接到左子树下方,然后将整个左子树作为右子树。
    在这里插入图片描述

如何把 root 的左右子树拉平?其实很简单,按照 flatten 函数的定义,对 root 的左右子树递归调用 flatten 函数即可:

public void flatten(TreeNode root) {
        if(root == null) return;
        flatten(root.left); //将左边的拉平
        flatten(root.right); //将右边的拉平

        /*
          后序遍历操作
        */
        TreeNode left = root.left; 
        TreeNode right = root.right;
        //将左边子树移到右边
        root.left = null;
        root.right = left;

        //原本的右边子树接在后面
        TreeNode p = root;
        while(p.right!=null){
            p = p.right;
        }
        p.right = right;

    }

flatten函数是怎么把左右子树拉平的?说不清楚,递归的解题就是不要陷入具体的细节,而是相信这个定义,让root去做它该做的事情。

复杂度分析
后更

相关推荐
©️2020 CSDN 皮肤主题: 1024 设计师:白松林 返回首页