剑指 Offer -- 二叉搜索树与双向链表(二十六)

二叉搜索树与双向链表(二十六)

题目描述

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

代码(已在牛客上 AC)

详见注释, 使用递归求解, 将右子树转换好之后, 更新根节点与右子树生成的链表的头结点之间的关系; 而为了更新左子树形成的链表与根节点之间的关系, 需要沿着左子树形成的链表一直走到链表结尾, 更新链表结尾和根节点的关系.

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
// 如果根节点为空或者没有任何子节点,那么直接返回 root;
// 否则, 先转换左子树, 然后转换右子树, 转换的规则是:
// 树的最左端节点为起始节点, 树的最右端节点是结束节点, 现在的问题就是,
// 如何连接 root 与 left 和 right.
// 首先: right 是排好序的, 所以判断 right 不为空的话, 只要使用 root->right
// 指向 right, right->left 指向 root, 就能正确连接 root 与 right.
// 之后: 连接 root 与 left 稍微复杂一些, 需要先对 left 进行遍历, 当然这要求
// left 不为空, ptr 一直遍历到最后一个节点, 然后 ptr->right 指向 root, 
// root->left 指向 ptr. 这样就连接好了. 但是最后返回的时候, 要判断 left 是否为空,
// 如果为空, 那么就返回 root, 否则返回 left.
class Solution {
public:
    TreeNode* Convert(TreeNode* root) {
        if (!root || (!root->left && !root->right)) return root;
        auto left = Convert(root->left);
        auto right = Convert(root->right);
        if (right) {
            root->right = right;
            right->left = root;
        }
        if (left) {
            auto ptr = left;
            while (ptr->right) ptr = ptr->right;
            root->left = ptr;
            ptr->right = root;
        }
        return left ? left : root;
    }
};