卡码网编程基础课 | 洗盘子

问题描述

题目描述

在餐厅里，洗盘子的工作需要使用到栈这种数据结构。

假设你手里有一个盘子堆放区。现在需要模拟洗盘子的过程，每个盘子都有一个编号。 

盘子堆放区操作说明： 

1. 当操作为 1 时，表示从盘子堆放区拿走顶部的盘子清洗。
2. 当操作为 2 时，表示有未洗的盘子放入盘子堆放区。 

在一系列操作之后，你需要回答：下一个清洗的盘子编号？

输入描述

第一行有一个整数 n，代表初始盘子堆放区中盘子的数量为 n。 

第二行有 n 个整数，代表了盘子的编号，同时整数之间的顺序也代表了未洗盘子加入盘子堆放区的顺序。 

第三行为一个整数 m，代表接下来将会有 m 次操作。 

接下来一共有 m 行，代表共有 m 次操作。 

如果是操作 1，那么该行只会有一个数字 1，代表有一个盘子被拿走清洗。 

如果是操作 2，那么该行有两个数字，第一个数字 2 表示有未洗的盘子加入，第二个数字代表未洗的盘子编号。

输出描述

输出共一行，为下一个该清洗的盘子编号。 如果没有下一个该清洗的盘子，那么请输出 “All the dishes have been washed.”

输入示例

5
1001 1002 1003 1004 1005
3
1
1
2 1006

输出示例

1006

python解法

# 读取初始盘子数量
n = int(input())

# 读取盘子的编号列表
plates = list(map(int, input().split()))

# 读取操作次数
m = int(input())

# 循环处理每次操作
for _ in range(m):
    operation = input().split()
    opt = int(operation[0])

    # 操作1: 拿走并清洗盘子
    if opt == 1:
        if plates:
            cleaned_plate = plates.pop()
    # 操作2: 将未洗的盘子放到堆放区
    elif opt == 2:
        plate_number = int(operation[1])
        plates.append(plate_number)

# 判断盘子堆放区是否为空
print("All the dishes have been washed.") if not plates else print(plates[-1])

C++解法

#include <iostream>
#include <stack>

int main(){
    int n, m, opt, val, num;
    std::cin >> n;
    std::stack<int> st; // 构建一个栈结构
    while (n--){
        std::cin >> val;
        st.push(val);   // 将输入的盘子编号使用push操作推到栈中
    }
    
    std::cin >> m;
    while (m--){
        std::cin >> opt;
        // 判断栈是否为空
        if (opt == 1 && !st.empty())
            st.pop();
        else if(opt == 2){
            std::cin >> num;
            st.push(num);
        }
    }
    // 使用empty()操作判断是否为空栈
    if (st.empty())
        std::cout << "All the dishes have been washed." << std::endl;
    else
        std::cout << st.top() << std::endl; // 使用top()获取栈顶元素
}