PTA 1048 数字加密(Python3)

本题要求实现一种数字加密方法。首先固定一个加密用正整数 A，对任一正整数 B，将其每 1 位数字与 A 的对应位置上的数字进行以下运算：对奇数位，对应位的数字相加后对 13 取余——这里用 J 代表 10、Q 代表 11、K 代表 12；对偶数位，用 B 的数字减去 A 的数字，若结果为负数，则再加 10。这里令个位为第 1 位。

输入格式：

输入在一行中依次给出 A 和 B，均为不超过 100 位的正整数，其间以空格分隔。

输出格式：

在一行中输出加密后的结果。

输入样例：

1234567 368782971

输出样例：

3695Q8118

提交结果：

代码：

a, b = input().split()
length1 = len(a)
length2 = len(b)
s = ''
if length1 > length2:
    max1 = length1
    b = '0' * (length1 - length2) + b
else:
    max1 = length2
    a = '0' * (length2 - length1) + a
a = a[::-1]
b = b[::-1]
for i in range(max1):
    if (i + 1) % 2 == 1:
        a1 = (int(a[i]) + int(b[i])) % 13
        if a1 == 10:
            a1 = 'J'
        elif a1 == 11:
            a1 = 'Q'
        elif a1 == 12:
            a1 = 'K'
        else:
            a1 = str(a1)
        s += a1
    else:
        b1 = int(b[i]) - int(a[i])
        if b1 < 0:
            b1 += 10
        s += str(b1)
print(s[::-1])