天梯 L1-025 正整数A+B python解法

题的目标很简单，就是求两个正整数A和B的和，其中A和B都在区间[1,1000]。稍微有点麻烦的是，输入并不保证是两个正整数。

输入格式：

输入在一行给出A和B，其间以空格分开。问题是A和B不一定是满足要求的正整数，有时候可能是超出范围的数字、负数、带小数点的实数、甚至是一堆乱码。

注意：我们把输入中出现的第1个空格认为是A和B的分隔。题目保证至少存在一个空格，并且B不是一个空字符串。

输出格式：

如果输入的确是两个正整数，则按格式A + B = 和输出。如果某个输入不合要求，则在相应位置输出?，显然此时和也是?。

输入样例1：

123 456

输出样例1：

123 + 456 = 579

输入样例2：

22. 18

输出样例2：

? + 18 = ?

输入样例3：

-100 blabla bla...33

输出样例3：

? + ? = ?

​
# 按字符串格式接收输入的那一行数据
str = input()
# 按空格分割保存至数组
shu = str.split(' ')
# 两个状态码，0是正常数据，1是非正常数据
A = 0
B = 0

# 判断函数，对输入的数据进行判断，是否是0到9的纯数字
def pan(str):
    for i in str:
        if '0' <= i <= '9':
            pass
        else:
            return 1
    return 0

# 判断数组长度是否大于2，大于2则代表B是乱码
if len(shu) > 2:
    B = 1
elif pan(shu[1]) == 1:
    B = 1
# 确定数据是否是在正确范围内
elif int(shu[1]) > 1000 or int(shu[1]) < 1:
    B = 1

if pan(shu[0]) == 1:
    A = 1
elif int(shu[0]) > 1000 or int(shu[0]) < 1:
    A = 1

if A == 1 and B != 1:
    print('? + {} = ?'.format(int(shu[1])))
elif A != 1 and B == 1:
    print('{} + ? = ?'.format(int(shu[0])))
elif A == 1 and B == 1:
    print('? + ? = ?')
else:
    print('{} + {} = {}'.format(int(shu[0]), int(shu[1]), int(shu[0]) + int(shu[1])))

​