题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
/*
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)
{
if(pRootOfTree == nullptr)
return nullptr;
TreeNode* pLast = nullptr;//指向双向链表的尾节点
convertNode(pRootOfTree, &pLast);//将子树转为链表
TreeNode* pHead = pLast;
while(pHead != nullptr && pHead->left != nullptr)
pHead = pHead->left;//沿双向链表向前遍历找到头节点
return pHead;
}
void convertNode(TreeNode* pRoot, TreeNode** pLast)
{
if(pRoot == nullptr)
return;
TreeNode* pCur = pRoot;
if(pCur->left != nullptr)
convertNode(pCur->left, pLast);
pCur->left = *pLast; //将根节点插入左子树形成的
if(*pLast != nullptr) //链表的 末端
(*pLast)->right = pCur; //
*pLast = pCur;//更新当前链表的尾节点
if(pCur->right != nullptr)
convertNode(pCur->right,pLast);
}
};