在C中计算堆栈中元素的数量

#include <stdio.h>
#include <stdlib.h>

#define MAX_SIZE 100

// 定义堆栈结构
typedef struct {
    int data[MAX_SIZE];
    int top;
} Stack;

// 初始化堆栈
void initStack(Stack *stack) {
    stack->top = -1;
}

// 判断堆栈是否为空
int isEmpty(Stack *stack) {
    return stack->top == -1;
}

// 判断堆栈是否已满
int isFull(Stack *stack) {
    return stack->top == MAX_SIZE - 1;
}

// 将元素压入堆栈
void push(Stack *stack, int element) {
    if (isFull(stack)) {
        printf("堆栈已满，无法添加元素。\n");
        return;
    }
    stack->data[++(stack->top)] = element;
}

// 从堆栈中弹出元素
int pop(Stack *stack) {
    if (isEmpty(stack)) {
        printf("堆栈为空，无法弹出元素。\n");
        return -1;
    }
    return stack->data[(stack->top)--];
}

// 计算堆栈中元素的数量
int countElements(Stack *stack) {
    int count = 0;
    
    // 复制堆栈内容到一个临时堆栈
    Stack tempStack;
    initStack(&tempStack);
    while (!isEmpty(stack)) {
        int element = pop(stack);
        push(&tempStack, element);
        count++;
    }
    
    // 将元素重新放回原始堆栈
    while (!isEmpty(&tempStack)) {
        int element = pop(&tempStack);
        push(stack, element);
    }
    
    return count;
}

int main() {
    Stack stack;
    initStack(&stack);
    
    push(&stack, 10);
    push(&stack, 20);
    push(&stack, 30);
    push(&stack, 40);
    
    int elementCount = countElements(&stack);
    printf("堆栈中的元素数量为：%d\n", elementCount);
    
    return 0;
}