力扣LeetCode经典算法 用两个栈实现一个队列

数据结构（三十二）

学习数据结构与算法过程中的心得体会以及知识点的整理，方便我自己查找，也希望可以和大家一起交流。

—— 用两个栈实现一个队列 ——

1.题目描述

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例：

示例 1：

输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]

输出：
[null,null,3,-1]

示例 2：

输入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出：
[null,-1,null,null,5,2]

2.代码

c

typedef struct st{
    struct st *next;
    int value; 
}stack;

typedef struct {
    stack s1;
    stack s2;
} CQueue;


CQueue* cQueueCreate() {//初始化队列
    CQueue *CQ=(CQueue*)malloc(sizeof(CQueue));
    CQ->s1.next=NULL;
    CQ->s2.next=NULL;
    return CQ;
}

void cQueueAppendTail(CQueue* obj, int value) {//压入数据
    stack *temST=(stack*)malloc(sizeof(stack));
    temST->next = obj->s1.next;
    temST->value = value;
    obj->s1.next = temST;//更新头结点
}

int cQueueDeleteHead(CQueue* obj) {//弹出数据
    int bottom_num = -1;
    if(obj->s1.next == NULL)//若s1为空，则直接返回-1
        return -1;
    while(obj->s1.next !=NULL){//若s1不为空，则进入循环，每次循环将s1的栈顶数据弹入s2中
        if(obj->s1.next->next == NULL){//当s1仅剩余一个数据时，弹出数据
            bottom_num = obj->s1.next->value;
            obj->s1.next = obj->s1.next->next;
        }
        else{//若s1不止一个数据，则每次循环弹出栈顶元素进入s2
            stack *temp = obj->s1.next->next;
            obj->s1.next->next = obj->s2.next;
            obj->s2.next = obj->s1.next;
            obj->s1.next = temp;
        }
    }
    while(obj->s2.next!=NULL){//接下来将s2的数据再弹回s1中
        stack *temp  = obj->s2.next->next;
        obj->s2.next->next = obj->s1.next;
        obj->s1.next = obj->s2.next;
        obj->s2.next = temp;
    }
    return bottom_num;
}
void cQueueFree(CQueue* obj) {//释放节点内存
    free(obj);
}

/**
 * Your CQueue struct will be instantiated and called as such:
 * CQueue* obj = cQueueCreate();
 * cQueueAppendTail(obj, value);
 
 * int param_2 = cQueueDeleteHead(obj);
 
 * cQueueFree(obj);
*/

3.代码说明

首先从代码的注释说明不难看出测试用例的输入方式通过调用函数的方式，因此题目的示例中的["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]很明显对应相应的函数，这点弄懂了就能明白这题实现的主要方式。与此同时清楚了输入与输出的方式，那么我们只需要运用最简单直接的方法就可以实现这道题，创建两个栈，一个用来存储数据，当需要输出时，将这个栈的数据按顺序压入另一个栈中，取第一个栈的栈底数据或者第二个栈的栈顶数据即可完成题目要求。这次选用了很普通的算法，因此时间和空间的占用都会略微有点高，后续还可以进行改进。