【剑指Offer】二叉搜索树与双向链表 解题报告(Python)

【剑指Offer】二叉搜索树与双向链表 解题报告(Python)

标签(空格分隔): 剑指Offer


题目地址:https://www.nowcoder.com/ta/coding-interviews

题目描述:

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

解题方法

这个题还真的不好想,不过看到BST之后我们明白做法就是中序遍历。中序遍历是有序的。使用了一个指针来保存链表的最后的节点。每次中序遍历到了根节点,就把根节点添加到了中序遍历的链表的末尾。这样遍历结束就能得到了一个中序有序的链表。最后再利用末尾指针逆序找到头结点即可。

【剑指Offer】二叉搜索树与双向链表 解题报告(Python)_第1张图片

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, pRootOfTree):
        self.linkedlistLast = None
        self.convertNode(pRootOfTree)
        pHead = self.linkedlistLast
        while pHead and pHead.left:
            pHead = pHead.left
        return pHead

    def convertNode(self, root):
        if not root: return
        pcurr = root
        if pcurr.left:
            self.convertNode(pcurr.left)
        pcurr.left = self.linkedlistLast
        if self.linkedlistLast:
            self.linkedlistLast.right = pcurr
        self.linkedlistLast = pcurr
        if pcurr.right:
            self.convertNode(pcurr.right)

Date

2018 年 3 月 20 日 – 阳光正好,安心学习

你可能感兴趣的:(算法,牛客网,剑指offer)