【牛客剑指offer刷题】：Python：26.二叉搜索树与双向链表

二叉搜索树与双向链表

时间限制：1秒 空间限制：32768K 热度指数：232897
算法知识视频讲解

题目描述

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

解析

搜索二叉树：

任意节点x，其左子树中的key不大于x.key，其右子树中的key不小于x.key。

中序遍历搜索二叉树的结果就是升序数组。

法1.非递归

分步实现：
1.将搜索二叉树升序存在list里
2.将list拼接成双向链表

代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, pRootOfTree):
        # write code here
        if not pRootOfTree:
            return None
        
        # 1.遍历搜索二叉树
        p = pRootOfTree
        
        stack = []
        resStack = []
        
        while p or stack:
            if p:
                stack.append(p)
                p = p.left
            else:
                node = stack.pop()
                resStack.append(node)
                p = node.right
        
        # 2.拼接双向链表 
        resP = resStack[0]
        while resStack:
            top = resStack.pop(0)
            if resStack:
                top.right = resStack[0]
                resStack[0].left = top
        return resP

法2.递归版本

在中序遍历的过程中，增加拼接双向链表的部分

代码

1.首先来看中序遍历

class Solution:
    def __init__(self):
        self.array = []
    def midOrder(self, root):
        if not root:
            return self.array
        self.midOrder(root.left)
        self.array.append(root.val)
        self.midOrder(root.right)

2.其次，在self.array.append(root.val)部分增加双向链表部分

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

#
# 
# @param pRootOfTree TreeNode类 
# @return TreeNode类
#
class Solution:
    def Convert(self , pRootOfTree ):
        # write code here
        if not pRootOfTree:
            return None 
        if not pRootOfTree.left and not pRootOfTree.right:
            return pRootOfTree
            
        left = self.Convert(pRootOfTree.left)
        if left:
            p = left
            # 找到左边链表的最后一个
            while p.right:
                p = p.right 
            # 接上左链表
            p.right = pRootOfTree 
            pRootOfTree.left = p 
            
        right = self.Convert(pRootOfTree.right)
        if right:
            # 接上右链表
            pRootOfTree.right = right 
            right.left = pRootOfTree 
            
        return left if left else pRootOfTree 

参考：
https://www.nowcoder.com/questionTerminal/947f6eb80d944a84850b0538bf0ec3a5

https://blog.csdn.net/u010005281/article/details/79657259