6.1堆棧的使用

題目描述

堆棧是一種基本的數據結構。堆棧具有兩種基本操作方式，push 和 pop。Push一個值會將其壓入棧頂，而 pop 則會將棧頂的值彈出。現在我們就來驗證一下堆棧的使用。

輸入描述:

對于每組測試數據，第一行是一個正整數 n，0<n<=10000(n=0 結束)。而后的 n 行，每行的第一個字符可能是'P’或者'O’或者'A’；如果是'P’，后面還會跟著一個整數，表示把這個數據壓入堆棧；如果是'O’，表示將棧頂的值 pop 出來，如果堆棧中沒有元素時，忽略本次操作；如果是'A’，表示詢問當前棧頂的值，如果當時棧為空，則輸出'E'。堆棧開始為空。

輸出描述:

對于每組測試數據，根據其中的命令字符來處理堆棧；并對所有的'A’操作，輸出當時棧頂的值，每個占據一行，如果當時棧為空，則輸出'E’。當每組測試數據完成后，輸出一個空行。

代碼6.1

    #include <iostream>
    #include <stdio.h>
    #include <stack>
    using namespace std;
    stack<int> S; //定義一個堆棧
    int main()
    {
        int n;
        while( scanf("%d",&n) != EOF && n != 0 )
        {
            for(int i = 1; i <= n; i++){
                char t;
                cin>>t; //scanf會把回車當成輸入導致出錯
                if(t == 'P') //入棧
                {
                    int m;
                    scanf("%d", &m);
                    S.push(m);
                }else if(t == 'A') //詢問棧頂元素
                {
                    if(S.empty()){
                        printf("E\n");
                    }else{
                        printf("%d\n", S.top());
                    }
                }else if(t == 'O') //彈出棧頂元素
                {
                    if(!S.empty()) //棧非空
                        S.pop();
                }
            }
            printf("\n");
        }
        return 0;
    }