OpenJudge计算概论-简单算术表达式求值

/*=====================================
简单算术表达式求值
总时间限制: 1000ms 内存限制: 65536kB
描述
2位正整数的简单算术运算（只考虑整数运算），算术运算为：
+，加法运算
-，减法运算
*，乘法运算
/，除法运算
%，取余运算。
运算符前后可能有空格.
算术表达式的格式为：
运算数 运算符 运算数
请输出相应的结果。
输入
算术表达式，如：
32+64
输出
整形算数运算的结果（结果值不一定为2位数，可能多于2位或少于2位），例如
96
样例输入
32+64
样例输出
96
======================================*/

#include<stdio.h>
int main()
{
    char str[50],c;
    int a=0,b=0,res,i=0;
    int flag=1;
    //freopen("4.in","r",stdin);
    gets(str);
    while(str[i]!='\0')
    {
        if(str[i]>='0'&&str[i]<='9')
        {
            if(flag==1)
            {
                a=a*10+str[i]-'0';
            }
            else
            {
                b=b*10+str[i]-'0';
            }
        }
        else if(str[i]!=' ')
        {
            c=str[i];
            flag++;
        }
        i++;
    }
    switch(c)
    {
        case '+':
            res=a+b;
            break;
        case '-':
            res=a-b;
            break;
        case '*':
            res=a*b;
            break;
        case '/':
            if(b==0) return 0;
            res=a/b;
            break;
        case '%':
            if(b==0) return 0;
            res=a%b;
            break;
    }
    printf("%d\n",res);
    return 0;
}