zoukankan      html  css  js  c++  java
  • 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?

    /**
     * Definition for binary tree
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    private:
        void swap(int& a,int&b)
        {
            int tmp=a;
            a=b;
            b=tmp;
        }
        TreeNode* findlarge(TreeNode* root,int val)
        {
            if(root==NULL) return NULL;
            TreeNode* result=NULL;
            if(root->val>val) 
            {
                val=root->val;
                result=root;
            }
            TreeNode* p1=findlarge(root->left,val);
            TreeNode* p2=findlarge(root->right,val);
            if(p1!=NULL) result=p1;
            if(p2!=NULL)
            {
                if(result==NULL) result=p2;
                else 
                    if(result->val<p2->val) result=p2;
            }
            return result;
        }    
        TreeNode* findsmall(TreeNode* root,int val)
        {
            if(root==NULL) return NULL;
            TreeNode* result=NULL;
            if(root->val<val) 
            {
                val=root->val;
                result=root;
            }
            TreeNode* p1=findsmall(root->left,val);
            TreeNode* p2=findsmall(root->right,val);
            if(p1!=NULL) result=p1;
            if(p2!=NULL)
            {
                if(result==NULL) result=p2;
                else 
                    if(result->val>p2->val) result=p2;
            }
            return result;
        }
    public:
        void recoverTree(TreeNode *root) 
        {
            if(root==NULL) return;
            TreeNode* left=findlarge(root->left,root->val);
            TreeNode* right=findsmall(root->right,root->val);
            if(left!=NULL && right!=NULL)
            {        
                swap(left->val,right->val);
                return;
            }
            if(right!=NULL)
            {
                swap(root->val,right->val);
                return;
            }
            if(left!=NULL)
            {
                swap(root->val,left->val);
                return;
            }
            recoverTree(root->left);
            recoverTree(root->right);
        }
    };
  • 相关阅读:
    x8086汇编在显存中显示字符串
    x8086汇编实现dos清屏(clear screen)
    原创:根据题目要求,通过素数的方式判断一个小的字符串是否属于另一个大的字符串的子集
    python signal信号
    转:python signal信号
    python signal(信号)
    python问题记录
    Python语言and-or的用法
    perl6中的q/qq/qx/qqx
    upupw注入by pass
  • 原文地址:https://www.cnblogs.com/erictanghu/p/3759585.html
Copyright © 2011-2022 走看看