4.3OJ题：用栈实现队列

力扣https://leetcode-cn.com/problems/implement-queue-using-stacks/点击上面网址可查看原题

题目

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false

说明：

你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可

解答

首先我们先把栈的接口函数完善进去

typedef int STDataType;
typedef struct Stack{
    STDataType* a;
    int top;
    int capacity;
}ST;
void StackInit(ST* st){
    assert(st);
    st->a=NULL;
    st->top=st->capacity=0;
}
void StackPush(ST* st,STDataType x){
    assert(st);
    if(st->top==st->capacity){
        int newcapacity=st->capacity==0?4:st->capacity*2;
        STDataType* tmp=(STDataType*)realloc(st->a,sizeof(STDataType)*newcapacity);
        if(tmp==NULL){
            printf("realloc fail\n");
            exit(-1);
        }
        st->a=tmp;
        st->capacity=newcapacity;
    }
    st->a[st->top++]=x;
}
void StackPop(ST* st){
    assert(st);
    assert(st->top>0);
    st->top--;
}
bool StackEmpty(ST* st){
    assert(st);
    return st->top==0;
}
int StackSize(ST* ps) {
    assert(ps);
    return ps->top;
}
void StackDestroy(ST* st){
    assert(st);
    free(st->a);
    st->a=NULL;
    st->top=st->capacity=0;
}
STDataType StackTop(ST* st){
    assert(st);
    assert(st->top>0);
    return st->a[st->top-1];
}

然后再完善MyQueue的功能

typedef struct {
    ST popSt;
    ST pushSt;
} MyQueue;


MyQueue* myQueueCreate() {
    MyQueue* qt=(MyQueue*)malloc(sizeof(MyQueue));
    StackInit(&qt->popSt);
    StackInit(&qt->pushSt); 
    return qt;
}

void myQueuePush(MyQueue* obj, int x) {
    assert(obj);
    StackPush(&obj->pushSt,x);
}

int myQueuePop(MyQueue* obj) {
    assert(obj);
    assert(!StackEmpty(&obj->popSt)||!StackEmpty(&obj->pushSt));
   
    //如果popST中没有数据，将pushST的数据导过去;如果popST中有数据，直接出popST栈顶元素
    //popST的数据就符合先进先出的顺序了
    if(StackEmpty(&obj->popSt)){
        while(!StackEmpty(&obj->pushSt)){
            StackPush(&obj->popSt,StackTop(&obj->pushSt));
            StackPop(&obj->pushSt);
        }
    }
    STDataType x=StackTop(&obj->popSt);
    StackPop(&obj->popSt);
    return x;
}

int myQueuePeek(MyQueue* obj) {
    assert(obj);
    assert(!StackEmpty(&obj->popSt)||!StackEmpty(&obj->pushSt));
    if(StackEmpty(&obj->popSt)){
        while(!StackEmpty(&obj->pushSt)){
            StackPush(&obj->popSt,StackTop(&obj->pushSt));
            StackPop(&obj->pushSt);
        }
    }
    return StackTop(&obj->popSt);
}

bool myQueueEmpty(MyQueue* obj) {
    return StackEmpty(&obj->popSt)&&StackEmpty(&obj->pushSt);
}

void myQueueFree(MyQueue* obj) {
    StackDestroy(&obj->popSt);
    StackDestroy(&obj->pushSt);
    free(obj);
}