当前位置:   article > 正文

⭐算法入门⭐《二叉树》中等01 —— LeetCode 114. 二叉树展开为链表

⭐算法入门⭐《二叉树》中等01 —— LeetCode 114. 二叉树展开为链表

一、题目

1、题目描述

  给你二叉树的根结点 root ,请你将它展开为一个单链表:展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。展开后的单链表应该与二叉树 先序遍历 顺序相同。
  样例输入: root = [1,2,5,3,4,null,6]
  样例输出: [1,null,2,null,3,null,4,null,5,null,6]

2、基础框架

  • C语言 版本给出的基础框架代码如下:
void flatten(struct TreeNode* root){
}
  • 1
  • 2

3、原题链接

LeetCode 114. 二叉树展开为链表

二、解题报告

1、思路分析

  前序遍历:先访问根结点,再前序遍历左子树,最后前序遍历右子树。

2、时间复杂度

  由于需要把所有结点都遍历一遍,所以时间复杂度为 O ( n ) O(n) O(n)

3、代码详解

struct TreeNode *head, *tail;

void preorder(struct TreeNode* root) {   // (1)
    struct TreeNode* now;
    if(root) {
        now = (struct TreeNode*) malloc( sizeof(struct TreeNode) );
        now->val = root->val;
        now->left = NULL;
        now->right = NULL;

        if (tail == NULL) {              // (2)
            head = tail = now;
        }else {
            tail->right = now;
            tail = now;
        }
        preorder(root->left);            // (3)
        preorder(root->right);           // (4)
    }
} 

void flatten(struct TreeNode* root){
    if(root == NULL) {
        return ;                         // (5)
    }
    head = tail = NULL;
    preorder(root);
    root->val = head->val;               // (6)
    root->left = NULL;
    root->right = head->right;
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • ( 1 ) (1) (1) 先序遍历将所有结点串成一串,根为 head;
  • ( 2 ) (2) (2) 对于第一个遍历到的结点特殊处理;
  • ( 3 ) (3) (3) 再前序遍历左子树;
  • ( 4 ) (4) (4) 最后前序遍历右子树;
  • ( 5 ) (5) (5) 如果是空树,直接返回,无需处理;
  • ( 6 ) (6) (6) 遍历完毕以后将头结点做一个处理,否则 root 还是指向原先的 二叉树;

三、本题小知识

  递归时处理二叉树问题很好的手段。


四、加群须知

  相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」
  那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:

推荐阅读
相关标签