考研机试 堆栈的使用

堆栈的使用

描述
堆栈是一种基本的数据结构。堆栈具有两种基本操作方式，push 和 pop。其中 push一个值会将其压入栈顶，而 pop 则会将栈顶的值弹出。现在我们就来验证一下堆栈的使用。（注：本题有多组输入，可以参考该网址：https://www.nowcoder.com/discuss/276）
输入描述：
对于每组测试数据，第一行是一个正整数 n（0 < n <= 10000）。而后的 n 行，每行的第一个字符可能是’P’或者’O’或者’A’；如果是’P’，后面还会跟着一个整数，表示把这个数据压入堆栈；如果是’O’，表示将栈顶的值 pop 出来，如果堆栈中没有元素时，忽略本次操作；如果是’A’，表示询问当前栈顶的值，如果当时栈为空，则输出’E’。堆栈开始为空。
输出描述：
对于每组测试数据，根据其中的命令字符来处理堆栈；并对所有的’A’操作，输出当时栈顶的值，每个占据一行，如果当时栈为空，则输出’E’。

#include <stack>
#include <string>
#include <iostream>
using namespace std;
int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
    stack<int> stack1;
    for(int i=0;i<n;++i){
        char ch;
        cin>>ch;
        if(ch=='A'){
            if(stack1.empty()){
                printf("E\n");
            }else{
                printf("%d\n",stack1.top());
            }
        }else if(ch=='O'){
            if(!stack1.empty()){
                stack1.pop();
            }
        }else if(ch=='P'){
            int m;
            cin>>m;
            stack1.push(m);
        }
    }}
}