vc6++里编写一个少儿数学加减法运算的程序

输入两个正整数和运算符+或-
输出格式为
A运算符B=
输出样例：
3 2
+
输出样例：
3+2=
大一学生 试了一个小时了 怎么都不对 求求各位大佬救救孩子吧

#include <iostream>
using namespace std;
int main()
{
    int a,b;
    char c;
    cin>>a>>b;
    cin>>c;
    if(c=='+')
      cout<<a<<c<<b<<"="<<a+b;
    else
      cout<<a<<c<<b<<"-"<<a-b;
    return 0;
}

```c++
#include<iostream>
using namespace std;
int main(){
    char s;
    int a,b;
    cin>>a>>b>>s;
    if(s=='+'){
        cout<<a+b;
        return 0;
    }
    if(s=='-'){
        cout<<a-b;
        return 0;
    }
}

```