PTA L1-025：正整数A+B (python)

一、题目要求

二、参考代码

arr=input()
s=['','']
c=0
for i in arr:
    if i!=' ':
        s[0]+=i
        c+=1
    else:
        break
s[1]=arr[c+1:]
flag=[0,0]
for i in range(2):
    try:
        t=int(s[i])
        if t<1 or t>1000:
            flag[i]=1
    except:
         flag[i]=1       
if flag[0]==0:
    if flag[1]==0:
        print("%d + %d = %d"%(int(s[0]),int(s[1]),int(s[0])+int(s[1])))
    else:
        print("%d + ? = ?"%(int(s[0])))
else:
    if flag[1]==0:
         print("? + %d = ?"%(int(s[1])))
    else:
        print("? + ? = ?")

注意：B可能是空格，不能直接用input().split()