diff --git a/sumsquareddigits.py b/sumsquareddigits.py new file mode 100644 index 0000000..dc40d36 --- /dev/null +++ b/sumsquareddigits.py @@ -0,0 +1,29 @@ +DIGS = "0123456789ABCDEFGH" + + +def int2base(n, base): + if n == 0: + return DIGS[0] + + digits = [] + while n: + digits.append(DIGS[n % base]) + n = n // base + + digits.reverse() + return "".join(digits) + + +def SSDN(b, n): + n = int2base(n, b) + ans = 0 + for x in n: + ans += DIGS.index(x) ** 2 + return ans + + +p = int(input()) + +for _ in range(p): + k, b, n = map(int, input().split()) + print(f"{k} {SSDN(b, n)}")