Ruudukko

from copy import deepcopy
 
def syote():
    global n
    n = int(input())
    rivit = []
    for i in range(n):
        rivit.append([int(luku) for luku in input().split(" ")])
    return rivit
 
def luokartta():
    global n
    kartta = [[0 for i in range(n)] for j in range(n)]
    return kartta
 
def summaarivi(rivi):
    summa = 0
    for luku in rivi:
        summa = summa + luku
    return summa % 1000000007
 
def summaasarake(lista, sarake):
    summa = 0
    for rivi in lista:
        summa = summa + rivi[sarake]
    return summa % 1000000007
 
def loyda(luvut):
    global n
    sanakirja = {}
    for riviindeksi in range(n):
        for sarakeindeksi in range(n):
            luku = luvut[riviindeksi][sarakeindeksi]
            koord = (riviindeksi, sarakeindeksi)
            if luku in sanakirja:
                sanakirja[luku].append(koord)
            else:
                sanakirja[luku] = [koord]
    return sanakirja
 
def korvaa(sanakirja, kartta):
    uusikartta = deepcopy(kartta)
    for avain in sorted(sanakirja.keys()):
        for koord in sanakirja[avain]:
            uusikartta[koord[0]][koord[1]] = summaarivi(kartta[koord[0]]) + summaasarake(kartta, koord[1]) + 1
        kartta = deepcopy(uusikartta)
    return kartta
 
def main():
    luvut = syote()
    kartta = luokartta()
    sanakirja = loyda(luvut)
    kartta = korvaa(sanakirja, kartta)
    summa = 0
    for rivi in kartta:
        summa = summa + summaarivi(rivi)
    print(summa % 1000000007)
 
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: ACCEPTED

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