#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
// 链表节点结构体
typedef struct Node {
int data;
struct Node* next;
} Node;
// 栈结构体
typedef struct {
Node* top;
} Stack;
// 队列结构体
typedef struct {
Node* front;
Node* rear;
} Queue;
/*----------- 栈操作 -----------*/
void initStack(Stack* stack) {
stack->top = NULL;
}
bool isStackEmpty(Stack* stack) {
return stack->top == NULL;
}
void push(Stack* stack, int data) {
Node* newNode = (Node*)malloc(sizeof(Node));
if (!newNode) {
fprintf(stderr, "内存分配失败\n");
exit(EXIT_FAILURE);
}
newNode->data = data;
newNode->next = stack->top;
stack->top = newNode;
}
int pop(Stack* stack) {
if (isStackEmpty(stack)) {
fprintf(stderr, "栈下溢错误\n");
exit(EXIT_FAILURE);
}
Node* temp = stack->top;
int data = temp->data;
stack->top = temp->next;
free(temp);
return data;
}
/*----------- 队列操作 -----------*/
void initQueue(Queue* queue) {
queue->front = NULL;
queue->rear = NULL;
}
bool isQueueEmpty(Queue* queue) {
return queue->front == NULL;
}
void enqueue(Queue* queue, int data) {
Node* newNode = (Node*)malloc(sizeof(Node));
if (!newNode) {
fprintf(stderr, "内存分配失败\n");
exit(EXIT_FAILURE);
}
newNode->data = data;
newNode->next = NULL;
if (isQueueEmpty(queue)) {
queue->front = queue->rear = newNode;
} else {
queue->rear->next = newNode;
queue->rear = newNode;
}
}
int dequeue(Queue* queue) {
if (isQueueEmpty(queue)) {
fprintf(stderr, "队列下溢错误\n");
exit(EXIT_FAILURE);
}
Node* temp = queue->front;
int data = temp->data;
queue->front = temp->next;
if (queue->front == NULL) {
queue->rear = NULL; // 关键修正
}
free(temp);
return data;
}
/*----------- 内存释放 -----------*/
void freeStack(Stack* stack) {
while (!isStackEmpty(stack)) {
pop(stack);
}
}
void freeQueue(Queue* queue) {
while (!isQueueEmpty(queue)) {
dequeue(queue);
}
}
int main() {
// 正确初始化示例
Stack s;
initStack(&s); // 必须初始化!
push(&s, 1);
push(&s, 2);
push(&s, 3);
printf("出栈元素:%d\n", pop(&s));
printf("出栈元素:%d\n", pop(&s));
printf("出栈元素:%d\n", pop(&s));
Queue q;
initQueue(&q); // 必须初始化!
enqueue(&q, 1);
enqueue(&q, 2);
enqueue(&q, 3);
printf("出队元素:%d\n", dequeue(&q));
printf("出队元素:%d\n", dequeue(&q));
printf("出队元素:%d\n", dequeue(&q));
freeStack(&s);
freeQueue(&q);
return 0;
}