二叉搜索树与双向链表

题目描述

来源于 https://leetcode-cn.com/

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

为了让您更好地理解问题,以下面的二叉搜索树为例:

我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。

下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。

特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。

注意:本题与主站 426 题相同:https://leetcode-cn.com/problems/convert-binary-search-tree-to-sorted-doubly-linked-list/

注意:此题对比原题有改动。

解法:

对二叉树搜索树做中序遍历,得到的序列就是递增的。仅需要在遍历的过程中,把每个节点和前一个节点连起来。

先找出最终链表的头结点,头节点为最终的返回值。因为遍历的过程中会修改二叉树,因此该步骤要提前进行。

在遍历的过程中,使用前驱节点的 right 指向当前节点,当前节点的 left 指向前驱节点,然后更新前驱节点。

遍历完成后,得到了一个双向链表,此时再把 head 和 tail 连起来,就构成了循环链表。

class Solution {
public:
    Node* treeToDoublyList(Node* root) {
        if(!root) return nullptr;

        Node *head = root;
        while(head->left){
            head = head->left;
        }
        
        Node *pre = nullptr;
        traversal(root, &pre);

        Node* tail = pre;
        tail->right = head;
        head->left = tail;
        return head;
    }

    void traversal(Node* node, Node** pre){
        if(node == nullptr) return;
        traversal(node->left, pre);
        
        if(*pre != nullptr){
            (*pre)->right = node;
        }
        node->left = *pre;
        *pre = node;

        traversal(node->right, pre);
    }
};

有一种想法是,一开始就找到 tail,然后把初始的前驱节点设置为 tail,认为这样遍历完成后,链表自动构成了循环链表了。这是错误的想法,因为如果一开始就设置 tail->right = head,那么遍历二叉树的过程就无法终止了。