From f945afbaad1fa25508991c5f762e4378be6146ed Mon Sep 17 00:00:00 2001
From: Jethro Stapelbroek <ik@jethrostapelbroek.nl>
Date: Mon, 26 Jun 2023 23:04:43 +0200
Subject: [PATCH] Sum Squared Digits Function

---
 sumsquareddigits.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 sumsquareddigits.py

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)}")