C语言算式判断

题目：小明终于学习了100以内的加减法，但是他不知道他计算的得数是否正确，然而他妈妈没有时间帮他检查，于是乎他找到了你，希望你能帮帮他。
输入
一个100以内加减法的数学表达式
输出
如果小明算的正确，输出YES，如果小明算错了，那么第一行输出NO，第二行输出正确结果的表达式
样例输入 Copy
1+2=4
样例输出 Copy
NO
1+2=3

代码实现如下：

#include
int main()
{
    int a,c,e;
    char b,d;
    scanf("%d%c%d%c%d",&a,&b,&c,&d,&e);
        if(a+c==e||a-c==e)
        {
        printf("YES\n");
        }
        else
        {
        printf("NO\n");
        if(b=='+') printf("%d+%d%c%d\n",a,c,d,a+c);
        else if(b=='-') printf("%d-%d%c%d\n",a,c,d,a-c);
        }

    return 0;
}

运行结果如下：