【分解让复杂问题简单化】二叉搜索树与双向链表

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ghscarecrow/article/details/82052448

此题出自牛客网的剑指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;
    }
};

猜你喜欢

转载自blog.csdn.net/ghscarecrow/article/details/82052448