此题出自牛客网的剑指offer专题
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向
解题思路
对于这道题,我们可以采用两个引用left,pre来指向链表的最左端(即第一个节点)以及前一个节点。由于二叉搜索树的特点,想要将其转换成一个排序的双向链表,我们可以采用中序遍历的算法。
代码如下
Java版本
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
TreeNode pre = null;
TreeNode left = null;
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree == null){
return null;
}
Convert(pRootOfTree.left);//找到最左端子节点并对pre,left引用进行初始化后开始构建左子树的双向链表
pRootOfTree.left = pre;
if(pre!=null){
pre.right = pRootOfTree;
}
left = left==null? pRootOfTree:left;
pre = pRootOfTree;
Convert(pRootOfTree.right);//对右子树进行构建
return left;
}
}
C++版本
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
TreeNode* pre = nullptr;
TreeNode* left = nullptr;
public:
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree==nullptr)
{
return nullptr;
}
Convert(pRootOfTree->left);
pRootOfTree->left = pre;
if(pre!=nullptr)
{
pre->right = pRootOfTree;
}
left = left==nullptr? pRootOfTree:left;
pre = pRootOfTree;
Convert(pRootOfTree->right);
return left;
}
};