题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
代码实现(c++):
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
stack<TreeNode*> TreeStack;
TreeNode* preNode;
TreeNode* pCur=pRootOfTree;
bool isfrist=true;
while(pCur || !TreeStack.empty())
{
if(pCur->left)
{
TreeStack.push(pCur);
pCur=pCur->left;
}
else
{
if(isfrist)
{
pRootOfTree=pCur;
preNode=pCur;
isfrist=false;
}
else
{
preNode->right=pCur;
pCur->left=preNode;
preNode=pCur;
}
pCur=pCur->right;
while(!pCur && !TreeStack.empty())
{
pCur=TreeStack.top();
preNode->right=pCur;
pCur->left=preNode;
preNode=pCur;
TreeStack.pop();
pCur=pCur->right;
}
}
}
return pRootOfTree;
}
};