【题解】二叉搜索树与双向链表

二叉搜索树与双向链表

题目链接:二叉搜索树与双向链表

解题思路1:递归+中序遍历

首先题目最后要求的是一个的递增的双向链表,而二叉搜索树也是一类非常有特色的树,它的根节点大于所有左侧的节点,同时又小于所有右侧的节点,如果我们按照左中右去遍历这颗二叉树,恰巧得到的就是一个递增序列

题目同时要求不要创建新的节点,这样我们就需要在原有树上进行操作,树有左右节点指针,双向链表有前后两个指针,正好一一对应,我们修改指针指向,结合中序遍历,得到一颗递增的双向链表

代码如下:

cpp 复制代码
	TreeNode* head = nullptr;
	TreeNode* pre = nullptr;
    TreeNode* Convert(TreeNode* pRootOfTree) {
        if(pRootOfTree == nullptr) return nullptr;//递归的结束条件
		Convert(pRootOfTree->left);//递归到最左节点,是最小值
		if(pre == nullptr){
			//此时pRootOfTree是最左节点,是链表的head
			//初始化head和pre
			head = pRootOfTree;
			pre = pRootOfTree;
		}else{
			//pre是每一个pRootOfTree的前驱节点
			pre->right = pRootOfTree;
			pRootOfTree->left = pre;
			pre = pRootOfTree;
		}
		Convert(pRootOfTree->right);
		return head;
    }

解题思路2:非递归+栈

我们利用栈先进后出的特性,来模拟中序遍历出所有元素,先让所有左侧的元素进栈,再依次取出其父节点,再找该节点的右节点,将节点进行连接,连接方式和上一种思路一样

cpp 复制代码
    TreeNode* Convert(TreeNode* pRootOfTree) {
		if(pRootOfTree == nullptr) return nullptr;
        TreeNode* head = nullptr;
		TreeNode* pre = nullptr;
		stack<TreeNode*> s;
		while(pRootOfTree!=nullptr || !s.empty()){
			while(pRootOfTree!=nullptr){
				s.push(pRootOfTree);
				pRootOfTree = pRootOfTree->left;
			}
			pRootOfTree = s.top();
			s.pop();
			if(pre == nullptr){
				head = pRootOfTree;
				pre = pRootOfTree;
			}else{
				pre->right = pRootOfTree;
				pRootOfTree->left = pre;
				pre = pRootOfTree;
			}
			pRootOfTree = pRootOfTree->right;
		}
		return head;
    }
相关推荐
czxyvX10 分钟前
016-二叉搜索树(C++实现)
开发语言·数据结构·c++
蒟蒻的贤13 分钟前
leetcode链表
算法·leetcode·链表
执着25939 分钟前
力扣hot100 - 94、二叉树的中序遍历
数据结构·算法·leetcode
-dzk-42 分钟前
【代码随想录】LC 707.设计链表
数据结构·c++·算法·链表
you-_ling2 小时前
数据结构:3.栈和队列
数据结构
梵刹古音2 小时前
【C语言】 递归函数
c语言·数据结构·算法
代码游侠3 小时前
C语言核心概念复习(二)
c语言·开发语言·数据结构·笔记·学习·算法
you-_ling3 小时前
数据结构:5.哈希表
数据结构·散列表
鲨辣椒100864 小时前
二叉树代码变现——递归函数实现深度遍历
数据结构
2301_810730104 小时前
python第四次作业
数据结构·python·算法