腾讯09-20技术研究与数据分析 第四题 K进制

from string import ascii_uppercase, digits
strings = digits + ascii_uppercase

def change_base(n, k):
    all_str = strings[:k]
    x,y = divmod(n,k)
    if x>0:
        return change_base(x,k) + all_str[y]
    else:             
        return all_str[y]

def get_origin(st, k):
    d = {i:j for i,j in zip(strings[:k], [i for i in range(0,k)])}
    s = 0
    for i, j in enumerate(list(st)[::-1]):
        s += d[j] * (k ** i)
    return s

if __name__ == '__main__':
    n = int(input())
    for i in range(n):
        k = int(input())
        str1, str2, chra = input().split()
        if chra == '+':
            print(change_base(abs(get_origin(str1, k) + get_origin(str2, k)), k))
        elif chra == '-':
            print(change_base(abs(get_origin(str1, k) - get_origin(str2, k)), k))
        elif chra == '*':
            print(change_base(abs(get_origin(str1, k) * get_origin(str2, k)), k))