题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
//直接用vector保存地址,即可,使用非递归的中序遍历
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree==NULL)return NULL;
if(pRootOfTree->left==NULL&&pRootOfTree->right==NULL)return pRootOfTree;
vector<TreeNode*> vec;
stack<TreeNode*> s;
TreeNode* cur=pRootOfTree;
while(s.empty()!=true||cur!=NULL)
{
while(cur!=NULL)
{
s.push(cur);
cur=cur->left;
}
auto m=s.top();
s.pop();
vec.push_back(m);
if(m->right!=NULL)
{
cur=m->right;
}
}
vec[0]->left=NULL;
vec[0]->right=vec[1];
vec[vec.size()-1]->left=vec[vec.size()-2];
vec[vec.size()-1]->right=NULL;
for(int i=1;i<vec.size()-1;++i)
{
vec[i]->left=vec[i-1];
vec[i]->right=vec[i+1];
}
return vec[0];
}
};