[剑指offer]二叉搜索树与双向链表

二叉搜索树与双向链表

题目

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

要求不能创建任何新的结点，只能调整树中结点指针的指向。

注意

• 需要返回双向链表最左侧的节点。

样例

输入下图中左边的二叉搜索树，则输出右边的排序双向链表。

题解

通过中序遍历构造单链表，然后，当单链表构造完成后，再构造双向链表

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* dummy, *p;
    void dfs(TreeNode* root) {
        if (root) {
            dfs(root->left);
            p->right = root, p = p->right;
            dfs(root->right);
        }
    }

    TreeNode* convert(TreeNode* root) {
        dummy = new TreeNode(-1);
        p = dummy;
        dfs(root);
        
        auto q = dummy->right;
        TreeNode* pre = NULL;
        while (q) q->left = pre, pre = q, q = q->right;
        
        return dummy->right;
    }
};