题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
二叉搜索树满足每个节点大于其左子节点(子树)小于其右子节点(子树),这就有了排序的关系,需要按照中序遍历的方法,从最左边的叶节点开始建立双向关系,当左边转换完之后,将根节点连接到其后,接下来转换右子树...对于每一个子树,也可以采用同样的方法,因此可以使用递归。
代码如下:
/*
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)
{
TreeNode* ListEnd = nullptr;
ConvertList(pRootOfTree, &ListEnd);
TreeNode* ListHead = ListEnd;
while(ListHead != nullptr && ListHead->left != nullptr)
ListHead = ListHead->left;
return ListHead;
}
void ConvertList(TreeNode* Node, TreeNode** ListEnd)
{
if(Node == nullptr)
return;
TreeNode* cur = Node;
if(cur->left != nullptr)
ConvertList(cur->left, ListEnd); // 从最左子叶(最小值)开始转换链表
cur->left = *ListEnd;
if(*ListEnd != nullptr)
(*ListEnd)->right = cur; // 建立双向连接
*ListEnd = cur; // 转换完左子树,将当前根节点作为最后一位,接下来处理右子树
if(cur->right != nullptr)
ConvertList(cur->right, ListEnd);
}
};