堆栈的使用

一道acm的堆栈题，几乎让我复习了一晚上的数据结构，果断AC了，上代码

题目描述：

堆栈是一种基本的数据结构。堆栈具有两种基本操作方式，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

我的ac代码：

#include <stdio.h>
#include <stdlib.h>

struct stacknode
{
	int data;
	struct stacknode *next;
};

typedef struct
{
	struct stacknode *top;	//栈顶指针
}linkstack;

void initstack(linkstack *s);
void push(linkstack *s, int x);
void pop(linkstack *s);
void stacktop(linkstack *s);

int main()
{
	int n, i, x;
	char ch;

	while(scanf("%d",&n) != EOF)
	{
		//循环结束条件
		if(n <= 0 || n > 10000)
			break;
		//循环操作
		linkstack *s = malloc(sizeof(struct stacknode));
		initstack(s);
		for(i = 0; i < n; i ++)
		{
			getchar();
			scanf(" %c",&ch);
			switch(ch)
			{
				case 'P':
					//接受入栈数据
					scanf("%d",&x);
					push(s, x);
					break;
				case 'A':
					//输出栈顶元素
					stacktop(s);			
					break;
				case 'O':
					//出栈
					pop(s);
					break;
			}
		}
		printf("\n");
	}

	return 0;
}

void initstack(linkstack *s)
{
	s -> top = NULL;
}

void push(linkstack *s, int x)
{
	struct stacknode *p;
	p = malloc(sizeof(struct stacknode));
	if(p != NULL)
	{
		p -> data = x;
		p -> next = s -> top;
		s -> top = p;
	}
}

void pop(linkstack *s)
{
	struct stacknode *p = s -> top;
	if(p)
	{
		s -> top = p -> next;
		free(p);
	}
}

void stacktop(linkstack *s)
{
	int x;
	if(s -> top == NULL)
	{
		printf("E\n");
	}else
	{
		x = s -> top -> data;
		printf("%d\n",x);
	}
}