Ruudukko

# from numbers import map
# n = 1000
 
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 summat():
    global n
    return [0 for i in range(n)]
 
def loyda(luvut):
    global n
    # create a dictionary, with keys [1..n^2], value []
    sanakirja = dict([ (i+1, []) for i in range(n*n)] )
 
    for riviindeksi in range(n):
        for sarakeindeksi in range(n):
            luku = luvut[riviindeksi][sarakeindeksi]
            koord = (riviindeksi, sarakeindeksi)
            sanakirja[luku].append(koord)
 
    return sanakirja
 
def korvaa(sanakirja, kartta):
    rivisummat = summat()
    sarakesummat = summat()
    for avain in sorted(sanakirja.keys()):
        temp_result = []
        for koordi in sanakirja[avain]:
            temp_result.append(rivisummat[koordi[0]] + sarakesummat[koordi[1]] + 1)
 
        for koordiindeksi in range(len(sanakirja[avain])):
            kartta[sanakirja[avain][koordiindeksi][0]][sanakirja[avain][koordiindeksi][1]] = temp_result[koordiindeksi]
            rivisummat[sanakirja[avain][koordiindeksi][0]] = (rivisummat[sanakirja[avain][koordiindeksi][0]] + temp_result[koordiindeksi]) % 1000000007
            sarakesummat[sanakirja[avain][koordiindeksi][1]] = (sarakesummat[sanakirja[avain][koordiindeksi][1]] + temp_result[koordiindeksi]) % 1000000007
 
    return kartta, rivisummat
 
 
def main():
    global n
    luvut = syote()
    # luvut = map
    kartta = luokartta()
    sanakirja = loyda(luvut)
    kartta, rivisummat = korvaa(sanakirja, kartta)
    summa = summaarivi(rivisummat)
    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: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED