pat 正整数A+B （c语言实现）

#include<stdio.h>
#include<string.h>
#include<ctype.h>
int solve(char a[])
{
	int len=strlen(a);
	int sum=0;
	for(int i=0;i<len;i++)
	{
		if(a[i]>='0'&&a[i]<='9')
		sum=sum*10+a[i]-'0';
		else return 1001;
	}
	return sum;
}
int judge(char a[])
{
	int len=strlen(a);
	if(len>4)return 0;
	else if(a[0]=='-') return 0;
	else if(solve(a)>1000||solve(a)<1)return 0;
	else return 1;
}
main()
{
	char a[1000],b[1000];
	scanf("%s",a);
	getchar();
	gets(b);
	if(judge(a)&&judge(b)) printf("%d + %d = %d",solve(a),solve(b),solve(a)+solve(b));
	else if(judge(a)) printf("%d + ? = ?",solve(a));
	else if(judge(b)) printf("? + %d = ?",solve(b));
	else printf("? + ? = ?");
	
}