每日一题——Python代码实现PAT乙级1048数字加密

本题遇到的坑：题目未说明但需要注意补长

A,B = map(int,input().split())


def f2(A,B):
    A,B=int(A),int(B)
    C=(A+B)%13
    if C==10: C="J"
    elif C==11:C="Q"
    elif C==12: C="K"
    else:C=str(C)
    return C

def f3(A,B):
    A,B=int(A),int(B)
    #print(A,B)
    C=(B-A+10)%10
    #print(C)
    return str(C)

def encrypt(A,B):
    A,B,C=str(A),str(B),""
    la,lb=len(A),len(B)
    if la> lb:
        B="0"*(la-lb)+B
        lb=la
    else:
        A="0"*(lb-la)+A
        la=lb
    i=1
    while i<lb and i<la:
        #print("lb=",lb,"la=",la)
        C= f3(A[la-i-1],B[lb-i-1]) + f2(A[la-i],B[lb-i])+C
        i=i+2
        #print(i,la,lb)
    
    if i==la and i%2==1:
        C= f2(A[la-i],B[lb-i])+C
    if i==la and i%2==0:
        C=f3(A[la-i],B[lb-i])+C
    return C
print(encrypt(A,B))