JZ36 二叉搜索树与双向链表

 题目来源：二叉搜索树与双向链表_牛客题霸_牛客网 (nowcoder.com)

描述

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

 

数据范围：输入二叉树的节点数 0 \le n \le 10000≤n≤1000，二叉树中每个节点的值 0\le val \le 10000≤val≤1000
要求：空间复杂度O(1)O(1)（即在原树上操作），时间复杂度 O(n)O(n)。

注意:

1.要求不能创建任何新的结点，只能调整树中结点指针的指向。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode，有左右指针，其实可以看成一个双向链表的数据结构

4.你不用输出双向链表，程序会根据你的返回值自动打印输出

输入描述：

二叉树的根节点

返回值描述：

双向链表的其中一个头节点。

我的代码：

采用先序遍历。

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<TreeNode*> t;
    void mid_order(TreeNode* root){//中序遍历
        if(!root){
            return;
        }
        if(root->left) mid_order(root->left);
        t.push_back(root);
        if(root->right) mid_order(root->right);
        return;
    }
    TreeNode* Convert(TreeNode* pRootOfTree) {
        if(!pRootOfTree)
            return pRootOfTree;
        TreeNode* node=pRootOfTree;
        mid_order(node);//获得先序便利的结果
        
        for(int i=0;i<t.size()-1;i++){
            t[i]->right = t[i+1];
            t[i+1]->left = t[i];
        }
        t[0]->left=nullptr;
        t[t.size()-1]->right=nullptr;//将第一个节点的左节点和第二个节点的右节点设置为null
        return t[0];
    }
};