剑指Offer 36 | 二叉搜索树和双向链表

131 阅读1分钟

如果对数据结构不够了解,可以看看: 万字长文解说数据结构

剑指Offer系列

刷题仓库

36.二叉搜索树和双向链表

题目描述

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

思路 & 解答

主要是使用递归,设置一个headendhead只在第一次设置值,后面就是将endright指向当前node,将当前nodeleft指向end,把end指向当前的node

代码如下:

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}

public class Solution {
    public TreeNode head = null;
    public TreeNode end = null;

    public TreeNode Convert(TreeNode pRootOfTree) {
        invert(pRootOfTree);
        return head;
    }

    public void invert(TreeNode node) {
        if (node == null) {
            return;
        }
        invert(node.left);
        if (end == null) {
            head = node;
            end =node;
        }else{
            end.right = node;
            node.left = end;
            end =node;
        }
        invert(node.right);
    }
}

C++ 代码如下:

class Solution {
public:
    TreeNode *head;
    TreeNode *end;

    TreeNode *Convert(TreeNode *pRootOfTree) {
        invert(pRootOfTree);
        return head;
    }

    void invert(TreeNode *node) {
        if (node == NULL) {
            return;
        }
        invert(node->left);
        if (end == NULL) {
            head = node;
            end = node;
        } else {
            end->right = node;
            node->left = end;
            end = node;
        }
        invert(node->right);
    }
};

时间复杂度:O(N)。
空间复杂度:O(N),主要是递归的栈空间。

【作者简介】
秦怀,公众号【秦怀杂货店】作者,个人网站:aphysia.cn,技术之路不在一时,山高水长,纵使缓慢,驰而不息。

剑指Offer全部题解PDF

开源编程笔记