LC-恢复二叉搜索树

最新推荐文章于 2022-02-09 17:01:55 发布

vector6_

最新推荐文章于 2022-02-09 17:01:55 发布

阅读量158

点赞数 1

分类专栏： Leetcode 文章标签：数据结构算法二叉树

本文链接：https://blog.csdn.net/qq_41105501/article/details/109345419

版权

Leetcode 专栏收录该内容

137 篇文章 0 订阅

订阅专栏

###恢复二叉搜索树

//恢复二叉搜索树
/*二叉搜索树（BST）中的两个节点被错误地交换了，
请在不改变树的结构的情况下恢复这棵树。  */

 struct TreeNode {
     int val;
     TreeNode *left;
     TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };

class Solution {
public:
	
	TreeNode *node1 = nullptr;
	TreeNode* node2 = nullptr;
	TreeNode* pre = nullptr;
	void recoverTree(TreeNode *root) {
		if (root == nullptr)
			return;
		dfs(root);
		if(node1 && node2)
			swap(node1->val, node2->val);
	}

	void dfs(TreeNode* root)
	{
		if (root == nullptr)
			return;
		dfs(root->left);
		if (pre != nullptr && pre->val > root->val)
		{
			if (!node1)
				node1 = pre;
			node2 = root; 
		}
		pre = root;
		dfs(root->right);
	}
};