【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!

简介: 二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。

题目链接


LeetCode 99. 恢复二叉搜索树[1]

题目描述


二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下,恢复这棵树。

示例1

输入
:[1,3,null,null,2]  
1  
/ 
3
\   2
输出:
[3,1,null,null,2]   3  / 1  \   2

示例2

输入:[3,1,4,null,null,2]  3 / \1   4   /  2输出:[2,1,4,null,null,3]  2 / \1   4   /  3

进阶:

image.png

题解


先不考虑空间复杂度,因为二叉搜索树的中序遍历是单调递增的,所以我们只需要求出它的中序遍历。然后两个结点被调换过位置,等价于递增序列中两个数调换了位置。那么我们只需要找出序列中第一个逆序对(前一个数)和最后一个逆序对(后一个数)就行了,然后换回它俩的位置。

但是中序遍历无论使用递归实现还是栈实现,空间复杂度都是树的高度,不是常数。

想一下为什么我们需要用递归或者栈来实现中序遍历?因为从根结点开始进入左子树之后,遍历完了左子树还需要回到根结点,然后再进入右子树继续遍历。但是如果你没有栈,你就没法从左子树回到根结点了,因为左子树中没有结点能指到根结点。

我们需要解决的就是这个问题,这里我们引入 Morris 遍历算法

下图是一个二叉搜索树例子:image.png

我们用递归来做中序遍历,访问结点的顺序是 4212346567 (这个序列就是 Morris 遍历得到的序列)。注意到非叶子结点 246 会被访问两次,第一次是作为根结点进入左子树,第二次是递归返回了,中序遍历到自己了。

那么不用递归或者栈,我们怎么在第二次成功返回根结点呢?注意到叶子结点的左右儿子都是空的,所以可以利用它们来指向根结点。

用结点 4 举个例子。首先访问 4 ,然后遍历它的左子树。左子树的最后一个遍历的结点是 3 ,那么我们就把 3 的右儿子指向根结点 4 。这样遍历完左子树之后,还能通过 3 的右儿子回到根结点 4 ,接着继续遍历右子树。

所以在第一次访问根结点时,首先要找出它左子树中最右边的那个叶子结点,把它的右儿子指向根结点。然后才能放心地递归遍历左子树,不用担心回不去啦。

具体实现的时候还有一些细节,看代码更好理解,我都写在注释里面了。

代码


c++

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */classSolution {
public:
voidrecoverTree(TreeNode*root) {     
TreeNode*x=NULL, *y=NULL, *pre=NULL, *rightmost=NULL;  
while (root) {  
// 如果有左子树,就递归遍历左子树。 if (root->left) {    
rightmost=root->left;   
// 找出左子树的最右边一个叶子结点   while (rightmost->right&&rightmost->right!=root)
                {           
rightmost=rightmost->right;    
                }     
// 如果左子树最右边的叶子结点的右儿子是空的, // 那就说明根结点是第一次访问,那么就把它的右儿子指向根结点。              // 然后递归遍历左子树。if (rightmost->right!=root) {   
rightmost->right=root;  
root=root->left;          
// 否则的话说明根结点是第二次访问了,// 那就说明左子树已经递归完毕了,    // 那么就判断一下是否存在逆序对。  // 记得把左子树最右叶子结点的右儿子改回空指针。// 然后递归遍历右子树了。                    } else {            
if (pre&&pre->val>root->val) {    
if (x==NULL) x=pre;  
y=root;    
                    }           
rightmost->right=NULL;  
pre=root;       
root=root->right;     
                }          
// 如果没有左子树,那就直接遍历右子树,同时判断是否存在逆序对。                 } else {      
if (pre&&pre->val>root->val) {      
if (x==NULL) x=pre;      
y=root;         
                }             
pre=root;     
root=root->right; 
            }     
        }      
swap(x->val, y->val);  
    }
};

参考资料


[1]

LeetCode 99. 恢复二叉搜索树: https://leetcode-cn.com/problems/recover-binary-search-tree/

image.png

作者简介:godweiyang知乎同名华东师范大学计算机系硕士在读,方向自然语言处理与深度学习喜欢与人分享技术与知识,期待与你的进一步交流~


相关文章
|
12月前
|
算法 C++
【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!
【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!
|
3天前
|
算法 数据安全/隐私保护 计算机视觉
基于二维CS-SCHT变换和LABS方法的水印嵌入和提取算法matlab仿真
该内容包括一个算法的运行展示和详细步骤,使用了MATLAB2022a。算法涉及水印嵌入和提取,利用LAB色彩空间可能用于隐藏水印。水印通过二维CS-SCHT变换、低频系数处理和特定解码策略来提取。代码段展示了水印置乱、图像处理(如噪声、旋转、剪切等攻击)以及水印的逆置乱和提取过程。最后,计算并保存了比特率,用于评估水印的稳健性。
|
1天前
|
算法
MATLAB|【免费】融合正余弦和柯西变异的麻雀优化算法SCSSA-CNN-BiLSTM双向长短期记忆网络预测模型
这段内容介绍了一个使用改进的麻雀搜索算法优化CNN-BiLSTM模型进行多输入单输出预测的程序。程序通过融合正余弦和柯西变异提升算法性能,主要优化学习率、正则化参数及BiLSTM的隐层神经元数量。它利用一段简单的风速数据进行演示,对比了改进算法与粒子群、灰狼算法的优化效果。代码包括数据导入、预处理和模型构建部分,并展示了优化前后的效果。建议使用高版本MATLAB运行。
|
3天前
|
算法 计算机视觉
基于高斯混合模型的视频背景提取和人员跟踪算法matlab仿真
该内容是关于使用MATLAB2013B实现基于高斯混合模型(GMM)的视频背景提取和人员跟踪算法。算法通过GMM建立背景模型,新帧与模型比较,提取前景并进行人员跟踪。文章附有程序代码示例,展示从读取视频到结果显示的流程。最后,结果保存在Result.mat文件中。
|
3天前
|
资源调度 算法 块存储
m基于遗传优化的LDPC码OMS译码算法最优偏移参数计算和误码率matlab仿真
MATLAB2022a仿真实现了遗传优化的LDPC码OSD译码算法,通过自动搜索最佳偏移参数ΔΔ以提升纠错性能。该算法结合了低密度奇偶校验码和有序统计译码理论,利用遗传算法进行全局优化,避免手动调整,提高译码效率。核心程序包括编码、调制、AWGN信道模拟及软输入软输出译码等步骤,通过仿真曲线展示了不同SNR下的误码率性能。
9 1
|
3天前
|
存储 算法 数据可视化
基于harris角点和RANSAC算法的图像拼接matlab仿真
本文介绍了使用MATLAB2022a进行图像拼接的流程,涉及Harris角点检测和RANSAC算法。Harris角点检测寻找图像中局部曲率变化显著的点,RANSAC则用于排除噪声和异常点,找到最佳匹配。核心程序包括自定义的Harris角点计算函数,RANSAC参数设置,以及匹配点的可视化和仿射变换矩阵计算,最终生成全景图像。
|
3天前
|
算法 Serverless
m基于遗传优化的LDPC码NMS译码算法最优归一化参数计算和误码率matlab仿真
MATLAB 2022a仿真实现了遗传优化的归一化最小和(NMS)译码算法,应用于低密度奇偶校验(LDPC)码。结果显示了遗传优化的迭代过程和误码率对比。遗传算法通过选择、交叉和变异操作寻找最佳归一化因子,以提升NMS译码性能。核心程序包括迭代优化、目标函数计算及性能绘图。最终,展示了SNR与误码率的关系,并保存了关键数据。
16 1
|
3天前
|
算法 调度
考虑需求响应的微网优化调度模型【粒子群算法】【matlab】
考虑需求响应的微网优化调度模型【粒子群算法】【matlab】
|
3天前
|
运维 算法
基于改进遗传算法的配电网故障定位(matlab代码)
基于改进遗传算法的配电网故障定位(matlab代码)
|
3天前
|
算法 调度
基于多目标粒子群算法冷热电联供综合能源系统运行优化(matlab代码)
基于多目标粒子群算法冷热电联供综合能源系统运行优化(matlab代码)