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))