JZ26 二叉搜索树与双向链表

描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。如下图所示

因为二叉搜索树的概念就是左子树比根节点小，右子树比根节点大，如果要输出一个排序的链表。

就是树的左根右，也就是中序遍历。

然后除了中序遍历外，还需要注意的是双向链表。

另外题目需要注意的地方：

1.要求不能创建任何新的结点，只能调整树中结点指针的指向。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode，有左右指针，其实可以看成一个双向链表的数据结构

这道题目主要是有左右指针，所以可以看成是一个双向链表。

讨论组有一个思路接近的：（可是我看得懂，写不出QAQ）

先中序遍历，将所有的节点保存到一个列表中。对这个list[:-1]进行遍历，每个节点的right设为下一个节点，下一个节点的left设为上一个节点。

class Solution:
    def Convert(self, pRootOfTree):
        # write code here
        if not pRootOfTree:return 
        self.arr = []
        self.midTraversal(pRootOfTree)
        for i,v in enumerate(self.arr[:-1]):
            v.right = self.arr[i + 1]
            self.arr[i + 1].left = v
        return self.arr[0]

    def midTraversal(self, root):
        if not root: return
        self.midTraversal(root.left)
        self.arr.append(root)
        self.midTraversal(root.right)

 其中关于enumerate()函数的用法：

 还有另外一种：先遍历左子树，然后找到左子树的最右的那个结点(左子树中最大的树)，将它的右指针指向root，同时root的左指针也要指向它，然后遍历右子树，把它加到root的右指针上面。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, root):
        if not root:
            return None
        if not root.left and not root.right:
            return root
        
        # 将左子树构建成双链表，返回链表头
        left = self.Convert(root.left)
        p = left
        
        # 定位至左子树的最右的一个结点
        while left and p.right:
            p = p.right
        
        # 如果左子树不为空，将当前root加到左子树链表
        if left:
            p.right = root
            root.left = p
        
        # 将右子树构造成双链表，返回链表头
        right = self.Convert(root.right)
        # 如果右子树不为空，将该链表追加到root结点之后
        if right:
            right.left = root
            root.right = right
            
        return left if left else root