PTA 7-1 正整数A+B (15分)

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
    char a;
    int sum1=0,sum2=0,flag1=0,flag2=0;
    while((a=getchar())!=EOF&&a!=' ')
    {
        if(a<'0'||a>'9')
            flag1=1;
        sum1*=10;
        sum1+=a-'0';
        if(sum1>1000||sum1<1)
            flag1=1;
    }
    while((a=getchar())!=EOF&&a!='\n')
    {
        if(a<'0'||a>'9')
            flag2=1;
        sum2*=10;
        sum2+=a-'0';
        if(sum2>1000||sum2<1)
            flag2=1;
    }
    if(flag1==0&&flag2==0)
        printf("%d + %d = %d",sum1,sum2,sum1+sum2);
    else if(flag1==1&&flag2==1)
        printf("? + ? = ?");
    else if(flag1==1&&flag2==0)
        printf("? + %d = ?",sum2);
    else
        printf("%d + ? = ?",sum1);
    return 0;
}