题解 | #二叉搜索树与双向链表#

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    // 设置pre保存上一个节点
    // head保存头结点
    TreeNode* pre = nullptr;
    TreeNode* head = nullptr;
    void function(TreeNode* node) {
        if(!node)
            return;
        function(node->left);
        if(pre)
            pre->right  = node; // 令上一个节点的右节点是当前节点
        else
            head = node; // 初始化头结点
        node->left = pre; // 令当前节点的左节点是上一个节点
        pre = node; // 保存上一个节点，以双链表为序列
        function(node->right);
    }
    TreeNode* Convert(TreeNode* pRootOfTree) {
        if(!pRootOfTree)
            return nullptr;
        function(pRootOfTree);
        return head;
    }
};