LeetCode 99:Recover Binary Search Tree

Subscribe to see which companies asked this question

一、题目描述

一棵二叉树原本是二叉搜索树，但是其中有两个节点调换了位置，要求恢复原来的二叉搜索树

二、解题思路

//中序遍历二叉树，出现的节点的值会升序排序，如果有两个节点位置错了，那肯定会出现降序。
 //设置一个pre节点指针，记录当前节点中序遍历时的前节点，如果当前节点小于pre节点的值，说明需要调整次序。
 //如果在中序遍历时节点值出现了两次降序，第一个错误的节点为第一次降序时较大的节点，第二个错误节点为第二次降序时较小的节点。
 //比如，原来的搜索二叉树在中序遍历的节点值依次为{1,2,3,4,5}，如果因为两个节点位置错了而出现{1,5,3,4,2}，
 //第一次降序为5->3，所以第一个错误节点为5，第二次降序为4->2,所以第二个错误节点为2，将5和2换过来就可以恢复。
 class Solution {
 public:
	 TreeNode* mistake1;
	 TreeNode* mistake2;
	 TreeNode* pre=NULL;

	 void recoverTree(TreeNode* root) {
		 recursive_traversal(root);
		 if (mistake1 != NULL && mistake2 != NULL)
		 {
			 swap(mistake1->val, mistake2->val);
		 }
	 }

	 //递归中序遍历二叉树
	 void recursive_traversal(TreeNode* root){
		 if (root == NULL)
			 return;
		 if (root->left != NULL)
		 {
			 recursive_traversal(root->left);
		 }
		 if (pre != NULL && pre->val>root->val)
		 {
			 if (mistake1 == NULL)
			 {
				 mistake1 = pre;
				 mistake2 = root;
			 }
			 else
			 {
				 mistake2 = root;
			 }
		 }

		 pre = root;

		 if (root->right != NULL)
		 {
			 recursive_traversal(root->right);
		 }
	 }
 };