Ruudukko

from copy import deepcopy
import sys

def syote():
    n = int(input())
    rivit = []
    for i in range(n):
        rivit.append([int(luku) for luku in input().split(" ")])
    return rivit

def luokartta(luvut):
    kartta = [[0 for i in luvut] for j in luvut]
    return kartta

def summaarivi(rivi):
    summa = 0
    for luku in rivi:
        summa = summa + luku
    return summa

def summaasarake(lista, sarake):
    summa = 0
    for rivi in lista:
        summa = summa + rivi[sarake]
    return summa

def merkkijonoksi(lista):
    merkkijono = ""
    for rivi in lista:
        for luku in rivi:
            merkkijono = merkkijono + str(luku) + " "
    return merkkijono

def korvaa(luku, luvut, kartta):
    uusikartta = deepcopy(kartta)
    for riviindeksi in range(len(luvut)):
        for sarakeindeksi in range(len(luvut)):
            if luvut[riviindeksi][sarakeindeksi] == luku:
                uusikartta[riviindeksi][sarakeindeksi] = summaarivi(kartta[riviindeksi]) + summaasarake(kartta, sarakeindeksi) + 1
    return uusikartta

def main():
    luvut = syote()
    kartta = luokartta(luvut)
    for luku in range(len(luvut)**2):
        kartta = korvaa(luku + 1, luvut, kartta)
    summa = 0
    for rivi in kartta:
        summa = summa + summaarivi(rivi)
    print(summa)

if __name__ == "__main__":
    main()

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED