AcWing 49 二叉搜索树与双向链表

题目描述：

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

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

注意：

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

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

分析：

题目要求我们将BST的中序遍历序列练成一个双向链表。可以看见，BST的垂直投影恰好就是有序的，对于链表中任一个结点，其前驱结点就是中序序列的前一个元素，其后继结点就是中序序列的后一个元素。

对于任意一个BST的任意一个结点，比如10，其中序的前驱是8（左子树最右边的结点），后继是12（右子树的最左边结点），我们将其连成双链表，得到8 10 12，对于6，同样的办法，得到4 6 8，对于14，得到12 14 16，不知不觉中，我们已经连成了4 6 8 10 12 14 16。

为此，我们需要一个递归函数，来找到某个结点最左边和最右边的点。首先考虑边界情况，某结点左右孩子为空，则返回{node，node}，左孩子为空，返回{node，node->right.second}，右孩子为空，返回{node->left.first,node}。一般情况下，返回{node->left.node->right.second}。在递归过程中，顺便把双链表的构建工作给完成即可。

/**
 * 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* convert(TreeNode* root) {
        if(!root)   return NULL;
        auto sides = dfs(root);
        return sides.first;
    }
    pair<TreeNode*,TreeNode*> dfs(TreeNode* node){
        if(!node->left && !node->right) return {node,node};
        if(node->left && node->right){
            auto l = dfs(node->left),r = dfs(node->right);
            l.second->right = node,node->left = l.second;
            r.first->left = node,node->right = r.first;
            return {l.first,r.second};
        }
        if(node->left){
            auto l = dfs(node->left);
            l.second->right = node,node->left = l.second;
            return {l.first,node};
        }
        if(node->right){
            auto r = dfs(node->right);
            r.first->left = node,node->right = r.first;
            return {node,r.second};
        }
    }
};