Kayaks

def main():
    (n, m) = [int(i) for i in input().split()]

    arr = []
    mm = {}
    for i in range(n):
        row = input()

        found = False
        for i in range(1, n + 1):
            if str(i) in row:
                found = True
                break
        if found:
            #mm[i] = len(arr)
            mm[len(arr)] = i
            arr.append(row)
    
    res = {}

    for i, a in enumerate(arr):
        num = mm[i]

        f = a.find(str(num))

        res[i+1] = m-2 - (f+2)

    asd = [(mm[k-1], res[k]) for k in res.keys()]

    s = sorted(asd, key=lambda x:x[1])

    sij = 1
    t = {}
    t[s[0][0]] = 1
    for i in range(1, len(s)):
        prev = t[s[0][0]]

        ss = s[i]
        if ss[1] > s[i-1][1]:
            sij += 1
            t[ss[0]] = sij
        else:
            t[ss[0]] = sij

    
    for i in range(1, n):
        print(t[i])



if __name__ == "__main__":
    main()

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 6

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 7

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 8

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 9

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 10

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 11

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10

Test 12

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 53, in <module>
    main()
  File "input/code.py", line 48, in main
    print(t[i])
KeyError: 10