Не могу разобраться с задачей, получаю TL

import sys
 
debug = 0
if debug:
    test = 2
    sys.stdin = open('input.txt', 'r')
    sys.stdout = open('output.txt', 'w')
else:
    test = 1
 
INF = 100000000000
 
def main():
    h, m, n = [int(i) for i in input().split()]
    size = m * n * h + n * m + n
    gr = []
    mats = []
 
    start = 0
    end = 0
    opened = '.12'
    for hi in range(h):
        mats.append([])
        for mi in range(m):
            mats[hi].append([])
            for ni, e in enumerate(input()):
                mats[hi][mi].append(e)
                gr.append([])
                if e == 'o':
                    continue
 
                i = m * n * hi + n * mi + ni
                if e == '1':
                    start = i
                elif e == '2':
                    end = i
 
                if ni > 0 and mats[hi][mi][ni - 1] in opened:
                    gr[i - 1].append(i)
                    gr[i].append(i - 1)
                if mi > 0 and mats[hi][mi - 1][ni] in opened:
                    gr[i - n].append(i)
                    gr[i].append(i - n)
                if hi > 0 and mats[hi - 1][mi][ni] in opened:
                    gr[i - n * m].append(i)
 
        if hi < h - 1:
            input()
 
    q = [start]
    dist = [INF] * size
    dist[start] = 0
    used = [0] * size
    while len(q):
        v = q[0]
        q.pop(0)
        used[v] = 1
        for i in gr[v]:
            if used[i]: continue
            dist[i] = dist[v] + 1
            q.append(i)
 
    print(dist[end] * 5)
 
 
for _ in range(test):
    main()