Kyselyt

import math as m

def calc_mag_size(mag, base):
  c = 0
  for i in range(1, mag + 1):
    c += (base**i - base**(i-1)) * i
  return c

def get_mag(n, b):
  c = 0
  i = 1
  while c < n:
    c += (b**i - b**(i-1)) * i
    i += 1
  return i - 1

def locate_digit(n, b):
  magnitude = get_mag(n, b)
  n -= calc_mag_size(magnitude-1, b)
  i = b**(magnitude - 1) + m.ceil(n / magnitude) - 1
  n -= (i - b**(magnitude-1)) * magnitude
  return int(str(i)[n - 1])

q = int(input())
r = []
for i in range(0, q):
    r.append(int(input()))

for i in r:
    print(locate_digit(i, 10))

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 30, in <module>
    print(locate_digit(i, 10))
  File "input/code.py", line 22, in locate_digit
    return int(str(i)[n - 1])
IndexError: string index out of range