LintCode 378. Convert Binary Search Tree to Doubly Linked List

最新推荐文章于 2024-01-22 16:06:43 发布

sparksnail

最新推荐文章于 2024-01-22 16:06:43 发布

阅读量531

点赞数

CC 4.0 BY-SA版权

分类专栏： OJ-LintCode 算法-树

本文链接：https://blog.youkuaiyun.com/y12345678904/article/details/79314154

OJ-LintCode 同时被 2 个专栏收录

85 篇文章

订阅专栏

算法-树

51 篇文章

订阅专栏

本文介绍了一种将二叉搜索树转换为双向链表的方法。通过递归分治策略，首先处理根节点，然后分别处理左子树和右子树。需要注意的是，在连接根节点与左右子树时，不仅要确保根节点正确连接，还要确保子树内部的next和prev指针也正确地相互连接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

这里写图片描述

思路

分治，拆分子问题即可。
但是，有一个小坑啊啊啊！
root连接了左右的node之后，左右的node的next和pre也要连接上node啊！！调了半天ε=(´ο｀*)))
好困。

代码

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        this.val = val
        this.left, this.right = None, None
Definition of Doubly-ListNode
class DoublyListNode(object):

    def __init__(self, val, next=None):
        self.val = val
        self.next = self.prev = next
"""


class Solution:
    """
    @param: root: The root of tree
    @return: the head of doubly list node
    """
    def convert(self, root):
        if not root:
            return None, None
        rootListNode = DoublyListNode(root.val)
        leftHead, leftTail = self.convert(root.left)
        rightHead, rightTail = self.convert(root.right)
        rootListNode.prev = leftTail
        rootListNode.next = rightHead
        if leftTail:
            leftTail.next = rootListNode
        if rightHead:
            rightHead.prev = rootListNode
        if leftHead and rightTail:
            return leftHead, rightTail
        elif leftHead:
            return leftHead, rootListNode
        elif rightTail:
            return rootListNode, rightTail
        else:
            return rootListNode, rootListNode
    def bstToDoublyList(self, root):
        # write your code here
        return self.convert(root)[0]