Sword finger Offer: [day 1 stack and queue (simple)] --- > stack containing min function

Posted by steelerman99 on Tue, 15 Feb 2022 14:00:23 +0100

1, Title Description

To define the data structure of the stack, please implement a min function that can get the smallest element of the stack in this type. In this stack, the time complexity of calling min, push and pop is all. O ( 1 ) \rm{O(1)} O(1).

Example:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();      --> return-3.
minStack.pop();
minStack.top();      --> Return 0.
minStack.min();      --> return-2.

Tips:
The total number of calls of each function shall not exceed 20000


2, Train of thought analysis

Note: some contents and pictures in the thought analysis refer to the predecessors. Thank them for their selfless dedication

analysis

The complexity of push() and pop() functions of ordinary stack is O(1), while the min() function to obtain the minimum value of stack needs to traverse the whole stack, and the complexity is O(N).
This problem needs to reduce the complexity of min() function to O(1), which can be realized by establishing auxiliary stack. Stack 1 is used to store all elements. push() function, pop() function and top() function at the top of stack remain normal. The smallest element of stack 1 is always placed on the top of stack 2, so that the O(1) complexity of min() function can be realized.

Function design

push(x) function
① Stack 1 can be put into the stack normally
② For stack 2, it should be noted that the top of stack 2 must be the smallest element of stack 1. Each time you enter the stack, judge whether stack 2 is empty. If it is empty, press X into stack 2. If stack 2 is not empty, judge the size relationship between X and the top element of stack 2. If x ≤ the top element of stack 2, press X into stack 2.
pop() function
① Stack 1 can exit the stack normally
② For stack 2, it should be noted that after stack 1 leaves stack 1 element, the top element of stack 2 should still be the smallest element of stack 1. Each time the stack is out, the out of stack element is marked as y. if y is equal to the top element of stack 2, stack 2 also performs out of stack
top() function
Directly return to the top element of stack 1
min() function
Directly return to the top element of stack 2
Example:









3, Overall code

The overall code is as follows

#define N 20000

//Create two stacks. Stack 1 is used for normal stack in and stack out operations, and stack 2 is used to prevent the smallest element of stack 1 from being on the top of the stack
typedef struct {
    int *Stack1;
    int *Stack2;
    int top1;
    int top2;
} MinStack;

/** initialize your data structure here. */

MinStack* minStackCreate() {
    MinStack* MS=(MinStack*)malloc(sizeof(MinStack));
    MS->Stack1 = (int*)malloc(sizeof(int)*N);
    MS->Stack2 = (int*)malloc(sizeof(int)*N);
    MS->top1 = -1;
    MS->top2 = -1;
    return MS;
}

void minStackPush(MinStack* obj, int x) {
    obj->Stack1[++obj->top1] = x; //Stack 1 enters the stack normally
    if(obj->top2 == -1){   //If stack 2 is empty
        obj->Stack2[++obj->top2] = x; //Push x into stack 2, too
    }
    else{
        //If stack 2 is not empty, and x < = stack top element of stack 2
        if(x <= obj->Stack2[obj->top2]){
            //Press x into the top of stack 2
            obj->Stack2[++obj->top2] = x;
        }
    }

}

void minStackPop(MinStack* obj) {
    //Save the top element of stack 1, and top 1-1
    int a = obj->Stack1[obj->top1];
    obj->top1--;
    //If the top element of stack 1 is equal to the top element of stack 2, pop up the top element of stack 2, top2-1
    if(a==obj->Stack2[obj->top2]){
        obj->top2--;
    }
}

//Return to the top element of stack 1 normally
int minStackTop(MinStack* obj) {
    return obj->Stack1[obj->top1];
}

//Return to the top element of stack 2 normally
int minStackMin(MinStack* obj) {
    return obj->Stack2[obj->top2];
}

void minStackFree(MinStack* obj) {
    free(obj->Stack1);
    free(obj->Stack2);
    free(obj);
}

/**
 * Your MinStack struct will be instantiated and called as such:
 * MinStack* obj = minStackCreate();
 * minStackPush(obj, x);
 
 * minStackPop(obj);
 
 * int param_3 = minStackTop(obj);
 
 * int param_4 = minStackMin(obj);
 
 * minStackFree(obj);
*/

Run and verify

Topics: Algorithm data structure queue stack