题解 | 密码游戏

#输入的整数先是通过先余10取最后一个数字然后输入的整数除以10接着取下一个数字，
#接下来逆序输出integer列表，例如输入1234 输出为1 2 3 4
num = int(input())
integer = []
i = 0
while i < 4:
    integer.append(num%10)
    num /= 10
    num = int(num)
    i += 1
integer.reverse()
#给每个数字做相同的操作即先加3然后余9获得新数字，并交换第一个和第三个元素，第二个和第四个元素
i = 0
while i < 4:
    integer[i] = (integer[i]+3) % 9
    i += 1
integer[0],integer[2] = integer[2],integer[0]
integer[1],integer[3] = integer[3],integer[1]
#要求输出整数而不是列表。如果我不写if语句那么测试中有一个输入7168输出0214，那么输出的是214而不是0214导致报错
con = integer[0] * 1000 + integer[1] * 100 + integer[2] * 10 + integer[3] * 1
if con < 1000:
    print(f'0{con}')
else:
    print(con)