题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路1:递归、左中右的中序遍历
head作为要返回的头结点
首先问题是要得到一条链
那么可以先得到左子树的那条链
然后把这条链拼在当前结点左边
然后再得到右子树的那条链
然后再把右链拼在当前节点右边
/*
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;
//左
TreeNode* head = Convert(pRootOfTree->left);
//中:把左拼接到中
TreeNode* t = head;
if(t != NULL){
while(t->right != NULL){
t = t->right;
}
t->right = pRootOfTree;
pRootOfTree->left = t;
}
//右,并把右拼接到中
TreeNode* r = Convert(pRootOfTree->right);
if(r != NULL){
r->left = pRootOfTree;
pRootOfTree->right = r;
}
return head != NULL ? head : pRootOfTree;
}
};
思路2:递归、右中左的中序遍历
借用一个全局变量head存放这条链的头结点
先得到右子树的那条链
再让右链的头结点拼到当前节点
再得到左子树的那条链
这时奇妙的事情发生了,得左链这一步操作的时候,实际上就是不断执行把右边连到中间
于是return head就哦了
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
TreeNode* head = NULL;
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree == NULL) return NULL;
//右
Convert(pRootOfTree->right);
//中:把右拼接到中
if(head != NULL){
head->left = pRootOfTree;
pRootOfTree->right = head;
}
head = pRootOfTree;
//左
Convert(pRootOfTree->left);
return head;
}
};