SWUST OJ(1044)

顺序栈基本操作的实现

 #include <iostream>
 #include <cstdlib>
 using namespace std;

 typedef struct stackList
 {
     int *top, *base;
     int stack_size;
 }SqStack;

 void Init_stack(SqStack &s, int n)
 {
     s.base = (int*)malloc(n*sizeof(int));
     if (!s.base)
     {
         exit(-);
     }
     s.top = s.base;
     s.stack_size = n;
 }

 void Push(SqStack &s, int e)
 {
     if(s.top - s.base == s.stack_size)
     {
         exit(-);
     }
     *s.top++ = e;
 }

 void Pop(SqStack &s)
 {
     if (s.top == s.base)
     {
         exit(-);
     }
     s.top--;
 }

 int getTop(SqStack s)
 {
     if (s.base == s.top)
     {
         return -;
     }
     return *--s.top;
 }

 int main()
 {
     SqStack s;
     int n,c;
     cin>>n;
     Init_stack(s,n);

     for (int i = ; i < n; ++i)
     {
         cin>>c;
         Push(s,c);
     }
     int k;
     cin>>k;

     for (int i = ; i < k; ++i)
     {
         Pop(s);
     }

     int t = getTop(s);
     cout<<t;
     return ;
 }

简易方法：

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

 int main()
 {
     stack<int> s;
     int n, c, k;
     cin>>n;
     for (int i = ; i < n; ++i)
     {
         cin>>c;
         s.push(c);
     }

     cin>>k;
     for (int i = ; i < k; ++i)
     {
         s.pop();
     }

     if (s.empty())
     {
         cout<<-;
         return ;
     }

     cout<<s.top();
     return ;
 }