二叉搜索树与双向链表

来源:互联网 发布:云计算与人工智能 编辑:程序博客网 时间:2024/06/08 07:36

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

//这个问题往中序遍历比较好想,先用递归的中序遍历将遍历的节点存放到vector中//然后,遍历vector元素两次,将它改成双向链表/*struct TreeNode {    int val;    struct TreeNode *left;    struct TreeNode *right;    TreeNode(int x) :            val(x), left(NULL), right(NULL) {    }};*/class Solution {public:    vector<TreeNode*>res;    void traverse(TreeNode *root)    {       if(!root)           return ;        traverse(root->left);        res.push_back(root);        traverse(root->right);    }    TreeNode* adjust()    {       for(int i=0;i<res.size();++i)       {          res[i]->right = res[i+1];       }        res[res.size()-1]->right = nullptr;        for(int i=res.size()-1;i>=1;--i)        {          res[i]->left = res[i-1];        }        return res[0];    }    TreeNode* Convert(TreeNode* pRootOfTree)    {      if(!pRootOfTree)          return nullptr;        traverse(pRootOfTree);       return adjust();    }};