数据结构_stack

问题描述

一天，小 L 发现了一台支持一下操作的机器：
IN x：将整数 x 入栈
POP：将栈顶元素出栈
ASUB：出栈两个数，将两数差的绝对值入栈
COPY：将栈顶元素（如果有的话）复制一份，入栈
现在小 L 想知道经过给定的 n 次操作之后，栈内所有元素之和是多少。
Notice：这台机器会自动忽略不合法的操作。

★数据输入
第一行一个正整数 n，表示有 n 次操作。
接下来有 n 行，每行一个操作。
0<=n<=1000, 0<=x<=1000

★数据输出
输出这台机器在 n 次操作后， 栈内所有元素之和。

★样例

输入：

4

IN 10

IN 11

ASUB

COPY

输出

2

 

思路

　　用数组模拟栈。应注意非法忽略操作

 

code

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

#define MAXN 1001

int arr[MAXN]={0};
int len=0;

void in()
{
    int num;
    scanf("%d",&num);
    getchar();
    arr[len++]=num;
}

void pop()
{
    if(len>0)
        len--;
}

void asub()
{
    if(len>=2)
    {
        int num = arr[len-2]-arr[len-1];
        arr[len-2] = num>0 ? num : -num;
        len--;
    }
}

void copy()
{
    if(len>0)
    {
        arr[len] = arr[len-1];
        len++;
    }
}

int getSum()
{
    int i,sum=0;
    for(i=0;i<len;i++)
    {
        sum+=arr[i];
    }
    return sum;
}

//void _disAll()
//{
//    int i;
//    for(i=0;i<len;i++)
//        printf("%d ",arr[i]);
//    printf("\n");
//}

int main()
{    
    int i=0;
    int opnum=0;
    char op[10]={0};
    
    scanf("%d",&opnum);
    getchar();
    for(i=0;i<opnum;i++)
    {
        scanf("%s",op);
        if(strcmp(op,"IN")==0)
        {
            in();
        }
        else if(strcmp(op,"POP")==0)
        {
            pop();
        }
        else if(strcmp(op,"ASUB")==0)
        {
            asub();
        }
        else if(strcmp(op,"COPY")==0)
        {
            copy();
        }
        //else printf("OP ERROR");
//        _disAll();
    }
    printf("%d\n",getSum());
    
    return 0;
}