20-21_N2_v2

def dig(l):
    if ord(l) >= ord('A') and ord(l) <= ord('F'):
        return 10 + ord(l) - ord('A')
    return ord(l) - ord('0')

def from_16(s):
    ar = []
    for i in s:
        ar.append(dig(i))
    ar = ar[::-1]
    tot = 0
    for i in range(len(s)):
        tot += (16**i) * ar[i]
    return tot


def xor(a, b):
    return a ^ b

def shift(x):
    s = bin(x)
    t = s[len(s) - 1] + s[:len(s) - 1]
    t = t[3:]
    res = int(t, 2)
    return res

def K_shift(x, K):
    for i in range(K):
        x = shift(x)
    return x

def f(a, b, K):
    a = from_16(a)
    b = from_16(b)
    res =  xor(a,b)
    res = K_shift(res, K)
    return res


a = 'AB00'
b = '1234'

print("a b = ", from_16(a), from_16(b))
print(10 * 16**3 + 11 * 16**2)
print(16**3 + 2*16**2 + 3*16 + 4)

K = 0
must = from_16('5C9A')
print(f(a, b, K), must)


"""
a = '1A2B'
b = 'FFFF'
K = 4
must = from_16('4E5D')
print("a b = ", from_16(a), from_16(b))
print(16**3 + 10*16**2 + 2*16 + 11)
print(15 * (16**3 + 16**2 + 16 + 1))

print(f(a,b, K), must)
"""