题目:
二叉树搜索树转换成排序双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
法一:
- 对BST进行中序遍历,然后将每个节点的地址保存起来。
- 对每个节点进行连接。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void Inorder(TreeNode* root, vector<TreeNode*>& v)
{
if(root)
{
Inorder(root->left, v);
v.push_back(root);
Inorder(root->right, v);
}
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(nullptr == pRootOfTree)
return nullptr;
vector<TreeNode*> v;
Inorder(pRootOfTree, v);
if(v.size()==1)
return v[0];
for(size_t i = 1; i<v.size()-1;i++)
{
//前一个节点指向left
v[i]->left = v[i-1];
//后一个节点指向right
v[i]->right = v[i+1];
}
//以上处理了除第一个和最后一个节点
//处理第一个节点
v[0]->left = nullptr;
v[0]->right = v[1];
//处理最后一个节点
v[v.size()-1]->left = v[v.size()-2];
v[v.size()-1]->right = nullptr;
return v[0];
}
};
法二:
遍历首先从最左侧节点开始,借助中序遍历将节点的左右指针指向进行改变,最后遍历的后头结点为0。
cur->left = 中序遍历下cur的前一个节点
cur->right = 中序遍历下cur的后一个节点
二叉树如下:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void Convert(TreeNode* root, TreeNode*& prev)
{
if(nullptr == root)
return;
//对左子树进行递归遍历
Convert(root->left, prev);
//root的左指针域指向prev,但是本次递归的时候无法将root的右指针域的指向改变
root->left = prev;
if(prev)
prev->right = root;
prev = root;
Convert(root->right, prev);
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
//树为空直接返回
if(nullptr == pRootOfTree)
return nullptr;
//1 找链表的首节点--即二叉树中最左侧的节点
TreeNode* head = pRootOfTree;
while(head->left)
head = head->left;
//2 对二叉树进行中序遍历,改变节点左右指针的指向
TreeNode* prev = nullptr;
Convert(pRootOfTree, prev);
return head;
}
};