leetcode之Recover Binary Search Tree

原题如下:

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:
A solution using O( n ) space is pretty straight forward. Could you devise a constant space solution?

confused what "{1,#,2,3}" means? > read more on how binary tree is serialized on OJ.

思路一:在允许空间复杂度为O(n)的情况下,可以首先进行先序遍历得到递增序列,则该递增序列中有两个逆序(特殊情况有一个逆序),分别找出这两个逆序点(从前往后找到第一个逆序的前一个点,从后往前找第一个逆序的后一个点)进行值交换即可。

 void recoverTree0(TreeNode *root) {
		vector<TreeNode *>v;
		inOrder(root,v);
		int i = 0;
		for( i = 0; i < v.size(); i++){
			if(v[i]->val > v[i + 1]->val)
				break;
		}
		int j =v.size() - 1;
		for(;j >0; j--){
			if(v[j]->val < v[j - 1]->val)
				break;
		}
		int temp = v[i]->val;
		v[i]->val = v[j]->val;
		v[j]->val = temp;
    }
	void inOrder(TreeNode *root,vector<TreeNode *>&v){
		if(root == NULL)
			return;
		inOrder(root->left,v);	
		v.push_back(root);		
		inOrder(root->right,v);
	}	

思路二:在思路一的基础上,可以在不保存有序序列的情况下仅利用两个指针保存需要交换的节点,即在遍历的过程中用pre保存前一个节点,然后与当前节点比较,如果为第一个逆序,则让key1指向pre,key2指向当前遍历的节点root(此为应对仅有一个逆序的特殊情况),如果key1不为空,则为第二个逆序,key2指向当前节点root,此时即可提前结束遍历。

class Solution {
public:
     TreeNode *key1 ,*key2 ,*pre;
	void recoverTree(TreeNode *root) {
		key1 = key2 = pre = NULL;
		inOrder(root);
		swap(key1->val,key2->val);
		return;
	}
	void inOrder(TreeNode *root){
		if(root == NULL)
			return;
		inOrder(root->left);
		if(pre  && pre->val > root->val){
			if(key1 == NULL){
			    key1 = pre;	
			    key2 = root;
			}
			else{
				key2 = root;
				return;
			}
		}
		pre = root;
		inOrder(root->right);       
	}
};


在手写出数列进行分析的情况下上题的思路就比较容易理解了,另外才发现C++提供了交换两个数的方法swap,还是要熟悉STL才行啊,还有就是类成员变量不能直接初始化(因为此时还没有内存空间),可以在构造函数以及成员方法中进行初始化。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值