OpenJudge | 栈的基本操作

总时间限制: 1000ms 内存限制: 1000kB

描述

栈是一种重要的数据结构，它具有push k和pop操作。push k是将数字k加入到栈中，pop则是从栈中取一个数出来。

栈是后进先出的：把栈也看成横向的一个通道，则push k是将k放到栈的最右边，而pop也是从栈的最右边取出一个数。

假设栈当前从左至右含有1和2两个数，则执行push 5和pop操作示例图如下：

          push 5          pop

栈   1 2  ------->  1 2 5 ------>  1 2

现在，假设栈是空的。给定一系列push k和pop操作之后，输出栈中存储的数字。若栈已经空了，仍然接收到pop操作，

则输出error。

输入

第一行为m，表示有m组测试输入，m<100。

每组第一行为n，表示下列有n行push k或pop操作。（n<150）

接下来n行，每行是push k或者pop，其中k是一个整数。

（输入保证同时在栈中的数不会超过100个）

输出

对每组测试数据输出一行。该行内容在正常情况下，是栈中从左到右存储的数字，数字直接以一个空格分隔，如果栈空，则不作输出；但若操作过程中出现栈已空仍然收到pop，则输出error。

样例输入

2
4
push 1
push 3
pop
push 5
1
pop

样例输出

1 5
error

思路

使用一个栈来模拟即可。

值得注意的点

1. error是最后输出的，不是在输入pop检测到栈空的时候输出。
2. 栈空则不输出。

Code

C++ (AI写的)

#include <iostream>
#include <stack>
#include <string>
using namespace std;

int main() {
    int m;
    cin >> m;
    while (m--) {
        int n;
        cin >> n;
        stack<int> s;
        bool flag = true;
        while (n--) {
            string op;
            cin >> op;
            if (op == "push") {
                int k;
                cin >> k;
                s.push(k);
            } else {
                if (s.empty()) {
                    flag = false;
                } else {
                    s.pop();
                }
            }
        }
        if (flag) {
            stack<int> t;
            while (!s.empty()) {
                t.push(s.top());
                s.pop();
            }
            while (!t.empty()) {
                cout << t.top() << " ";
                s.push(t.top());
                t.pop();
            }
            cout << endl;
        } else {
            cout << "error" << endl;
        }
    }
    return 0;
}

C++ (我写的)

#include <bits/stdc++.h>
using namespace std;
int main() {
	array<int, 200> S;
	string op;
	int m, n, k, p = 0, q = 0;
	cin >> m;
	for(int i = 1; i <= m; i++) {
		cin >> n;
		p = 0;
		q = 0;
		for(int j = 1; j <= n; j++) {
			cin >> op;
			if(op == "push") {
				cin >> k;
				S[p++] = k;
			} else if(op == "pop") {
				if(p == 0) q = 1;
				else p--;
			}
		}
		if(n != 0 && q == 0) {
			for(int i = 0; i < p; i++) {
				if(i == 0) cout << S[i];
				else cout << " " << S[i];
			}
			cout << endl;
		} else if(q == 1) cout << "error" << endl;
	}
}

C++ (错误的)

#include <bits/stdc++.h>
using namespace std;
int main() {
	array<int, 200> S;
	string op;
	int m, n, k, p = 0;
	cin >> m;
	for(int i = 1; i <= m; i++) {
		cin >> n;
		p = 0;
		for(int j = 1; j <= n; j++) {
			cin >> op;
			if(op == "push") {
				cin >> k;
				S[p++] = k;
			} else if(op == "pop") {
				if(p == 0) cout << "error" << endl;
				else p--;
			}
		}
		for(int i = 0; i < p; i++) {
			if(i == 0) cout << S[i];
			else cout << " " << S[i];
		}
		if(n != 0 && p != 0) cout << endl;
	}
}