LeetCode(99)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?
분석은 다음과 같습니다.
첫 번째 방법이자 내가 현재 생각하고 있는 가장 좋은 방법은 O(n) 공간의 복잡도이다.바로 중서열을 두루 훑어보는 토대에서 두 원소가 한 조로 되어 순서가 뒤바뀌었는지 검사하는 것이다.
case1: 순서가 바뀌는 두 원소는 인접 원소이다
case2: 순서가 바뀐 두 원소는 인접 원소가 아니다
코드는 다음과 같습니다.
//400ms , , ?
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void recoverTree(TreeNode *root) {
        if(root==NULL)
            return;
        stack<TreeNode*> node_stack;
        while(root!=NULL){
            node_stack.push(root);
            root=root->left;
        }
        TreeNode* cur=NULL;
        TreeNode* pre=NULL;
        TreeNode* res_a=NULL;
        TreeNode* res_b=NULL;
        while(!node_stack.empty()) {
            cur=node_stack.top();
            node_stack.pop();
            if(pre!=NULL&&cur->val<=pre->val&&res_a==NULL){
                res_a=pre;
                res_b=cur;
            }else if(pre!=NULL&&cur->val<=pre->val&&res_a!=NULL) {
                res_b=cur;
            }
            pre=cur;
            cur=cur->right;
            while(cur!=NULL){
                node_stack.push(cur);
                cur=cur->left;
            }
        }
        int tmp=res_a->val;
        res_a->val=res_b->val;
        res_b->val=tmp;
        return;
    }
};

두 번째 방법은 상당히 총명하다.O(1)의 공간 복잡도를 실현했다.
분석을 해 보면 순서가 뒤바뀐 점을 찾으려면 반드시 중순으로 옮겨야 한다. 만약에 중순으로 옮겨가면 창고를 사용하든지 아니면 귀속하지 않든지.스택을 사용할 경우 반드시 O(N)의 공간 복잡도를 사용합니다.만약 귀속되지 않는다면, 적어도 개조의 희망은 있다.그래서 문제는 돌아가는 중순을 개조하여 그 공간의 복잡도를 O(1)로 바꾸고 순서가 뒤바뀐 점을 찾을 수 있게 하는 것으로 바뀌었다.코드는 다음과 같습니다.
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *first;
    TreeNode *second;
    TreeNode *pre;
     
    void inOrder(TreeNode *root){
        if (root==NULL)
            return;
        inOrder(root->left);
        if (pre == NULL){pre = root;}
        else {
            if (pre->val > root->val){
                if (first==NULL) {first = pre;}
                second = root;
            }
            pre = root;
        }
        inOrder(root->right);
    }
    void recoverTree(TreeNode *root) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        pre = NULL;
        first = NULL;
        second= NULL;
        inOrder(root);
        int val;
        val = first->val;
        first->val=second->val;
        second->val=val;
        return;
    }
};

좋은 웹페이지 즐겨찾기