{ }codelper
栈P0348

食堂的餐盘堆

食堂里有一个叠放餐盘的盘堆,开始时是空的。
餐盘上写着编号,可能有多个餐盘编号相同。
依次给出 q 条命令。
put x 把一个编号为 x 的餐盘放到盘堆的最上面。
take c 表示 c 名学生依次各拿走盘堆最上面的一个餐盘。
如果盘堆里的餐盘不到 c 个,就有多少拿多少,其余学生空手回去。
对每条 take 命令,在一行中按拿走的顺序输出被拿走的餐盘编号,用一个空格隔开。
如果一个餐盘也没拿到,就在这一行输出 -。
处理完所有命令后,在最后一行从最上面往下依次输出盘堆里剩下的餐盘编号,用一个空格隔开。
如果没有剩下的餐盘,就在最后一行输出 -。

输入
第一行是 q(1 ≤ q ≤ 9,000),接下来 q 行每行一条命令。
1 ≤ x ≤ 1,000,000,1 ≤ c ≤ 9,000(都是整数,只给出数字,不带逗号)。
命令的单词和数用一个空格隔开
输出
每条 take 命令一行,最后一行是剩下的餐盘(编号只输出数字,不带逗号)
样例 1 · stdin
8
put 3
put 7
put 5
take 2
take 3
take 1
put 4
put 9
stdout
5 7
3
-
9 4
main.py
输入(stdin)

运行时这些行会依次作为 input()/标准输入传入(运行中不会等待输入)。

$ 就绪

请在电脑上打开以编写和运行代码。在手机上可以查看题目和题解。

其他语言怎么写?查看各语言题解 · 8 种语言题解

放到最上面、从最上面取出的栈,各语言用的东西不一样。
Python 用列表的 append、pop(),JS、TS 用数组的 push、pop(),末尾就是最上面。
Java 用 ArrayDeque 的 push、pop,C# 用 Stack<int> 的 Push、Pop。
C++ 的 std::stack 不能遍历内部,所以为了输出最后一行要用 vector 的 push_back、pop_back。
C 用数组加一个表示最上面位置的数来实现。
Craykon 用列表的 添加 放入,用 按位置移除(数量 - 1) 取出。
陷阱一:从空栈取出时发生的事情不一样。
Python 的 pop() 因 IndexError 停止,Java 的 ArrayDeque.pop() 因 NoSuchElementException 停止,C# 的 Pop() 因 InvalidOperationException 停止。
示例的 take 3 出现在只剩一个餐盘的时候,所以不先问是否为空就取三次,程序会停在那里。
C++ 对空 vector 做 pop_back,以及 C 里最上面的位置小于 0,会发生什么是没有规定的。
陷阱二:JS 的 pop() 在空数组上也不报错,只返回 undefined。
如果把它也存进去,明明什么都没拿到,列表却不是空的,于是示例的 take 1 输出的是空行而不是 -(join 把 undefined 写成空字符串)。
在 TS 里 pop() 的类型是 number | undefined,直接放进 number[] 会被报告为类型错误。
陷阱三:在 Python 里用 pile[-c:] 一次切出来很方便,但切片是从下面那一侧开始排的。
示例的第一行会变成 7 5 而不是 5 7。
陷阱四:从头到尾遍历栈时的顺序也不一样。
Java 的 ArrayDeque 和 C# 的 Stack<int> 从最上面开始遍历,而 Java 旧的 Stack 类从最下面开始,示例的最后一行会变成 4 9。
Python 列表、JS 数组、C++ 的 vector、Craykon 列表也是从最下面开始,所以要倒着遍历或者先反转(reversed(pile)、pile.reverse()、rbegin()、反转())。

Pythonmain.py · 18 行
import sys lines = sys.stdin.read().split("\n")q = int(lines[0])pile = []  # 리스트의 끝이 더미의 맨 위out = []for i in range(1, q + 1):    cmd, v = lines[i].split()    v = int(v)    if cmd == "put":        pile.append(v)    else:        taken = []        while len(taken) < v and pile:  # 비었는지 먼저 묻고 꺼낸다            taken.append(pile.pop())        out.append(" ".join(map(str, taken)) if taken else "-")out.append(" ".join(map(str, reversed(pile))) if pile else "-")  # 맨 위부터print("\n".join(out))
JavaScriptmain.js · 18 行
const lines = require("fs").readFileSync(0, "utf8").split("\n");const q = Number(lines[0]);const pile = []; // 배열의 끝이 더미의 맨 위const out = [];for (let i = 1; i <= q; i++) {  const [cmd, t] = lines[i].split(" ");  const v = Number(t);  if (cmd === "put") {    pile.push(v);  } else {    const taken = [];    // 비었는지 먼저 묻고 꺼낸다(빈 배열의 pop()은 오류 없이 undefined를 준다)    while (taken.length < v && pile.length > 0) taken.push(pile.pop());    out.push(taken.length > 0 ? taken.join(" ") : "-");  }}out.push(pile.length > 0 ? pile.reverse().join(" ") : "-"); // 맨 위부터console.log(out.join("\n"));
TypeScriptmain.ts · 24 行
import { readFileSync } from "node:fs"; const lines: string[] = readFileSync(0, "utf8").split("\n");const q: number = Number(lines[0]);const pile: number[] = []; // 배열의 끝이 더미의 맨 위const out: string[] = [];for (let i = 1; i <= q; i++) {  const [cmd, t] = lines[i].split(" ");  const v: number = Number(t);  if (cmd === "put") {    pile.push(v);  } else {    const taken: number[] = [];    // pop()의 타입은 number | undefined: 비었는지 먼저 묻고 꺼낸다    while (taken.length < v) {      const top: number | undefined = pile.pop();      if (top === undefined) break;      taken.push(top);    }    out.push(taken.length > 0 ? taken.join(" ") : "-");  }}out.push(pile.length > 0 ? pile.reverse().join(" ") : "-"); // 맨 위부터console.log(out.join("\n"));
JavaMain.java · 41 行
import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;import java.util.ArrayDeque;import java.util.Deque; public class Main {    public static void main(String[] args) throws IOException {        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));        int q = Integer.parseInt(br.readLine().trim());        Deque<Integer> pile = new ArrayDeque<>(); // push·pop이 맨 위에서 일어난다        StringBuilder sb = new StringBuilder();        for (int i = 0; i < q; i++) {            String[] parts = br.readLine().split(" ");            int v = Integer.parseInt(parts[1]);            if (parts[0].equals("put")) {                pile.push(v);            } else {                int count = 0;                while (count < v && !pile.isEmpty()) { // 비었는지 먼저 묻고 꺼낸다                    if (count > 0) sb.append(' ');                    sb.append(pile.pop());                    count++;                }                if (count == 0) sb.append('-');                sb.append('\n');            }        }        if (pile.isEmpty()) {            sb.append('-');        } else {            boolean first = true;            for (int x : pile) { // ArrayDeque는 맨 위부터 돈다                if (!first) sb.append(' ');                sb.append(x);                first = false;            }        }        System.out.println(sb);    }}
C#Main.cs · 37 行
using System;using System.Collections.Generic;using System.Text; class Program{    static void Main()    {        int q = int.Parse(Console.ReadLine()!);        var pile = new Stack<int>();        var sb = new StringBuilder();        for (int i = 0; i < q; i++)        {            string[] parts = Console.ReadLine()!.Split(' ');            int v = int.Parse(parts[1]);            if (parts[0] == "put")            {                pile.Push(v);            }            else            {                int count = 0;                while (count < v && pile.Count > 0) // 비었는지 먼저 묻고 꺼낸다                {                    if (count > 0) sb.Append(' ');                    sb.Append(pile.Pop());                    count++;                }                if (count == 0) sb.Append('-');                sb.Append('\n');            }        }        // Stack<T>는 맨 위부터 돈다        sb.Append(pile.Count > 0 ? string.Join(" ", pile) : "-");        Console.WriteLine(sb);    }}
C++main.cpp · 37 行
#include <iostream>#include <string>#include <vector>using namespace std; int main() {    ios::sync_with_stdio(false);    cin.tie(nullptr);    int q;    cin >> q;    vector<int> pile; // std::stack은 속을 돌 수 없어서 vector를 스택으로 쓴다(끝이 맨 위)    for (int i = 0; i < q; i++) {        string cmd;        int v;        cin >> cmd >> v;        if (cmd == "put") {            pile.push_back(v);        } else {            int count = 0;            while (count < v && !pile.empty()) { // 비었는지 먼저 묻고 꺼낸다                if (count > 0) cout << ' ';                cout << pile.back();                pile.pop_back();                count++;            }            if (count == 0) cout << '-';            cout << '\n';        }    }    if (pile.empty()) cout << '-';    for (auto it = pile.rbegin(); it != pile.rend(); ++it) { // 맨 위부터        if (it != pile.rbegin()) cout << ' ';        cout << *it;    }    cout << '\n';    return 0;}
Cmain.c · 33 行
#include <stdio.h>#include <string.h> int pile[9000]; /* pile[top - 1]이 더미의 맨 위 */ int main(void) {    int q, top = 0;    if (scanf("%d", &q) != 1) return 0;    for (int i = 0; i < q; i++) {        char cmd[8];        int v;        if (scanf("%7s %d", cmd, &v) != 2) return 0;        if (strcmp(cmd, "put") == 0) {            pile[top++] = v;        } else {            int count = 0;            while (count < v && top > 0) { /* 비었는지 먼저 묻고 꺼낸다 */                if (count > 0) printf(" ");                printf("%d", pile[--top]);                count++;            }            if (count == 0) printf("-");            printf("\n");        }    }    if (top == 0) printf("-");    for (int i = top - 1; i >= 0; i--) { /* 맨 위부터 */        printf("%d", pile[i]);        if (i > 0) printf(" ");    }    printf("\n");    return 0;}
Craykonmain.cray · 37 行
q = 控制台.读行().转数字();pile = []; // 목록의 끝이 더미의 맨 위循环 (i = 0; i < q; i++){    parts = 控制台.读行().分割();    v = parts[1].转数字();    如果 (parts[0] == "put")    {        pile.添加(v);    }    否则    {        taken = [];        当 (taken.数量 < v && pile.数量 > 0)        {            taken.添加(pile[pile.数量 - 1]);            pile.按位置移除(pile.数量 - 1);        }        如果 (taken.数量 == 0)        {            控制台.输出("-");        }        否则        {            控制台.输出(taken.连接(" "));        }    }}如果 (pile.数量 == 0){    控制台.输出("-");}否则{    pile.反转(); // 맨 위부터    控制台.输出(pile.连接(" "));}