二叉搜索树与双向链表(C++)

题目:

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

思路:

二叉搜索树满足每个节点大于其左子节点(子树)小于其右子节点(子树),这就有了排序的关系,需要按照中序遍历的方法,从最左边的叶节点开始建立双向关系,当左边转换完之后,将根节点连接到其后,接下来转换右子树...对于每一个子树,也可以采用同样的方法,因此可以使用递归。

代码如下:

/*
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)
    {
        TreeNode* ListEnd = nullptr;
        ConvertList(pRootOfTree, &ListEnd);
        TreeNode* ListHead = ListEnd;
        while(ListHead != nullptr && ListHead->left != nullptr)
            ListHead = ListHead->left;
        return ListHead;
    }
    void ConvertList(TreeNode* Node, TreeNode** ListEnd)
    {
        if(Node == nullptr)
            return;
        TreeNode* cur = Node;
        if(cur->left != nullptr)
            ConvertList(cur->left, ListEnd);  // 从最左子叶(最小值)开始转换链表
        cur->left = *ListEnd;
        if(*ListEnd != nullptr)
            (*ListEnd)->right = cur; // 建立双向连接
        *ListEnd = cur; // 转换完左子树,将当前根节点作为最后一位,接下来处理右子树
        if(cur->right != nullptr)
            ConvertList(cur->right, ListEnd);
    }
};
发布了63 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Jerry_Leo_liu/article/details/105033163
今日推荐