题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
(1)、非递归
我们知道,二叉排序树用中序遍历的方法访问出来的序列刚好就是有序序列。因此可以采用中序遍历的非递归算法来解决这个问题。
以上图的二叉排序树为例:先一直让根节点及根节点的左子树进栈,10 5 4,此时可以修改4节点的指针指向。4节点出栈。则访问5节点出栈,修改5节点的指向,7节点进栈,修改7节点的指向,再出栈,修改10节点的指向,接着12节点进栈,修改指向,遍历完成后栈为空了。结束中序遍历。
实现:
/*
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==NULL)
{
return NULL;
}
stack<TreeNode*>s;
TreeNode *p=pRootOfTree;//遍历树的,初始指向根节点
TreeNode *before=NULL;//指向上一个被遍历的节点,初始指向NULL
TreeNode *head=NULL;//用于返回的双向链表的第一个
bool isfrist = true;//判断是否是左子树最左的节点
while(p!=NULL||!s.empty())
{
while(p!=NULL)
{
s.push(p);
p = p->left;
}
p = s.top();//栈顶指针
s.pop();
if(isfrist)//是最左边的指针
{
head = p;
before = head;
isfrist = false;
}
else{
before->right = p;
p->left = before;
before = p;
}
p = p->right;
}
return head;
}
(2)、递归(发现树的好多问题都要用到递归啊。。。)
将左子树构成双向链表,并返回左子树的头结点
将左链表定位到最后一个节点。
若左链表不为空,则将当前的root追加到左链表的后面。
将右子树构成双向链表并返回右链表的头结点
右链表不为空,则将它追加到root的后面
左链表为空,返回root,不为空返回左链表头结点。
实现:
/*
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==NULL)
{
return NULL;
}
if(pRootOfTree->left==NULL&&pRootOfTree->right==NULL)
{
return pRootOfTree;
}
TreeNode *left = Convert(pRootOfTree->left);//左子树的链表头结点
if(left!=NULL)//左链表不为空则将root追加到左链表后面
{
TreeNode *p = left;//用于寻找左链表的最后一个节点
while(p->right!=NULL)
{
p = p->right;
}
pRootOfTree->left = p;
p->right = pRootOfTree;
}
TreeNode *right = Convert(pRootOfTree->right);//将右子树合成右链表并返回头结点
if(right!=NULL)//右链表不空,追加到root的后面
{
pRootOfTree->right = right;
right->left = pRootOfTree;
}
return left!=NULL?left:pRootOfTree;
}
};
finished!!!