HJ50 四则运算

描述

输入一个表达式（用字符串表示），求这个表达式的值。

保证字符串中的有效字符包括[‘0’-‘9’],‘+’,‘-’, ‘*’,‘/’ ,‘(’， ‘)’,‘[’, ‘]’,‘{’ ,‘}’。且表达式一定合法。

数据范围：表达式计算结果和过程中满足  ，字符串长度满足 

输入描述：

输入一个算术表达式

输出描述：

得到计算结果

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int priority(char s)
{
    //规则：右括号 < + - < * / < 左括号
    int s_a = 0;
    if(s == '(' || s == '[' || s == '{')
        s_a = 3;
    else if(s == '*' || s == '/')
        s_a = 2;
    else if(s == '+' || s == '-')
        s_a = 1;
    
    return s_a;
}

int fun(int a, int b, char op)
{
    if(op == '+')
        return a + b;
    if(op == '-')
        return a - b;
    if(op == '*')
        return a * b;
    if(op == '/')
        return a / b;
    return 0;
}

int main()
{
    char str[1000];
    scanf("%s",str);

    int len = strlen(str);
    int s1[1000] = {};//两个栈，一个栈用来记录数字，一个栈用来记符号
    char s2[1000];

    int top1 = 0;
    int top2 = 0;
    for(int i = 0; i < len; i++)
    {
        int n = 0;
        if(str[i] >= '0' && str[i] <= '9')
        {
            while(str[i] >= '0' && str[i] <= '9')
            {
                n = 10 * n + str[i] - '0';
                i++;
            }
            i--;
            s1[top1] = n;
            top1++;
        }

        else if(str[i] == '-' && (str[i - 1] == '('||str[i - 1] == '['||str[i - 1] == '{'||i == 0))
        {
            //如果遇到- 应该判断是负号还是减号，如果-前没有数字则是负号
            i++;
            s1[top1] = str[i]- '0';
            s1[top1] = 0 - s1[top1];
            top1++;
        }
        else
        {
            if(str[i] == '}' || str[i] == ']' || str[i] == ')')  //如果遇到右括号应该一直计算到栈顶是左括号
            {
                while(s2[top2 - 1] != '{' && s2[top2 - 1] != '[' && s2[top2 - 1] != '(')
                {
                    int b = s1[top1 - 1];
                    int a = s1[top1 - 2];
                    //s1栈顶两个数字出栈
                    char op = s2[top2 - 1];
                    top2--; //s2栈顶符号出栈
                    s1[top1 - 2] = fun(a, b, op);
                    top1--;
                }
                top2--;
                continue;
            }

            int p1 = priority(str[i]);
            int p2 = priority(s2[top2 - 1]);
            if(top2 == 0 || p1 > p2 || s2[top2 - 1] == '{' || s2[top2 - 1] == '[' || s2[top2 - 1] == '(' ) //栈是空的，或者遇到了更高级的运算符，直接进栈,左括号不能忽略
            {
                s2[top2] = str[i];
                top2++;
            }
            else if(p1 <= p2)  //如果栈顶是左括号忽略
            {
                int b = s1[top1 - 1];
                int a = s1[top1 - 2];
                //s1栈顶两个数字出栈
                char op = s2[top2 - 1];
                top2--; //s2栈顶符号出栈
                s1[top1 - 2] = fun(a, b, op);
                top1--;
                i--;
            }
        }
    }//进栈完毕了
    top1--;
    top2--;
    while(top1>0)  //栈1不空的时候
    {
        int b = s1[top1];
        int a = s1[top1 - 1];
        char op = s2[top2];
        s1[top1 - 1] = fun(a, b, op);
        top2--;
        top1--;
    }
    printf("%d", s1[0]);
    return 0;
}