PTA 7-1 多二了一点 (15 分)

题目

若一个正整数有 2n 个数位，后 n 个数位组成的数恰好比前 n 个数位组成的数多 2，则称这个数字“多二了一点”。如 24、6668、233235 等都是多二了一点的数字。

给定任一正整数，请你判断它有没有多二了那么一点。

输入格式：
输入在第一行中给出一个正整数 N（≤10
1000
）。

输出格式：
在一行中根据情况输出下列之一：

如果输入的整数没有偶数个数位，输出 Error: X digit(s)，其中 X 是 N 的位数；
如果是偶数位的数字，并且是多二了一点，输出 Yes: X - Y = 2，其中 X 是后一半数位组成的数，Y 是前一半数位组成的数；
如果是偶数位的数字，但并不是多二了一点，输出 No: X - Y != 2，其中 X 是后一半数位组成的数，Y 是前一半数位组成的数。

输入样例 1：
233235
结尾无空行
输出样例 1：
Yes: 235 - 233 = 2
结尾无空行
输入样例 2：
5678912345
结尾无空行
输出样例 2：
No: 12345 - 56789 != 2
结尾无空行
输入样例 3：
2331235
结尾无空行
输出样例 3：
Error: 7 digit(s)
结尾无空行

解题思路

inputStr = str(input())
# inputStr = str("233235")

if len(inputStr)%2 != 0:
    print("Error: "+str(len(inputStr))+" digit(s)")
else:
    length = len(inputStr)
    A1 = inputStr[:length//2]
    A2 = inputStr[length//2:]
    # print(A1,A2)
    A1Res,A2Res = 0,0
    for i in A1:
        A1Res += int(i)
    for i in A2:
        A2Res += int(i)
    # print(A1Res,A2Res)
    if (A2Res - A1Res) == 2:
        print("Yes: {} - {} = 2".format(A2, A1))
    else:
        print("No: {} - {} != 2".format(A2, A1))