堆棧的使用

題目描述：

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

輸入：

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

輸出：

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

樣例輸入：

3
A
P 5
A
4
P 3
P 6
O 
A
0

樣例輸出：

E
5

3

題不難，但是容易出錯，涉及字符的讀入；

#include<stdio.h>
#include<stack>
using namespace std;
/*
仔細看題目中這句話：而後的 n 行，每行的第一個字符可能是'P’或者'O’或者'A’；
所以不能用getchar只能用scanf。用scanf讀字符串
*/
int main()
{
	int n;
	char c[2];
	while(scanf("%d",&n)!=EOF)
	{
		if(n==0)
			break;
		stack<int> s;
		while(n--)
		{
			scanf("%s",&c);
			if(c[0]=='P')
			{
				int num;
				scanf("%d",&num);
				s.push(num);
			}
			else if(c[0]=='O')
			{
				if(!s.empty())
					s.pop();
			}
			else if(c[0]=='A')
			{
				if(!s.empty())
					printf("%d\n",s.top());
				else
					printf("E\n");
			}
		}
		printf("\n");
	}

	return 0;
}

/*AC*/
#include<iostream>
#include<stack>
#include<cstdio>
using namespace std;
int main()
{
        int n;
        while(scanf("%d",&n)&&n != 0)
        {
                while(getchar() != '\n');
                char ch;
                int tmp;
                stack<int> ss;
                for(int i = 0;i < n;i++)
                {
                        scanf("%c",&ch);
                        switch(ch)
                        {
                        case 'P':
                                scanf("%d",&tmp);
                                ss.push(tmp);
                                break;
                        case 'O':
                                if(!ss.empty()) ss.pop();
                                break;
                        case 'A':
                                if(ss.empty()) printf("E\n");
                                else printf("%d\n",ss.top());
                                break;
                        }
                        while(getchar() != '\n');
                }
                puts("");
        }
        return 0;
}