Ruudukko

import java.util.ArrayList;
import java.util.Scanner;

/**
 *
 * @author NM
 */
public class Ruudukko {

    public static void main(String[] args) {
        Scanner lukija = new Scanner(System.in);
        int koko = lukija.nextInt();
        Long[][] numerot = new Long[koko][koko];
        Long[][] siirtojenMaara = new Long[koko][koko];
       // System.out.println("KOKO: " + koko);
        for (int y = 0; y < koko; y++) {
            for (int x = 0; x < koko; x++) {
                numerot[x][y] = lukija.nextLong();
                siirtojenMaara[x][y] = (long) 0;
            }
        }

        boolean ensimmainenIollut = false;
        boolean ensimmainenIollutApu = false;

//tarvitaan modulo
        //System.out.println("ALKAA");
        for (int i = 1; i <= koko * koko; i++) {
            // Integer[][] siirtojenMaaraKopio = new Integer[koko][koko];
            ArrayList<Long> pystyRivit = new ArrayList<>();
            ArrayList<Long> vaakaRivit = new ArrayList<>();
            for (int kopiointiY = 0; kopiointiY < koko; kopiointiY++) {
                long rivinSumma = 0;
                for (int kopiointiX = 0; kopiointiX < koko; kopiointiX++) {
                    rivinSumma = rivinSumma + siirtojenMaara[kopiointiX][kopiointiY];

                }
                vaakaRivit.add(kopiointiY, rivinSumma);
            }
            for (int kopiointiX = 0; kopiointiX < koko; kopiointiX++) {
                long pystySumma = 0;
                for (int kopiointiY = 0; kopiointiY < koko; kopiointiY++) {
                    pystySumma = pystySumma + siirtojenMaara[kopiointiX][kopiointiY];

                }
                pystyRivit.add(kopiointiX, pystySumma);
            }
            //System.out.println("VAAKA: " + vaakaRivit);
            // System.out.println("PYSTY: "+pystyRivit);
            for (int y = 0; y < koko; y++) {
                for (int x = 0; x < koko; x++) {
                    if (numerot[x][y] == i && ensimmainenIollut) {

                        //lasketaan pystysumma
                        siirtojenMaara[x][y] += pystyRivit.get(x) + vaakaRivit.get(y) + 1;

                        //lasketaan vaakasumma
                    } else if (numerot[x][y] == i && ensimmainenIollut == false) {
                        siirtojenMaara[x][y] = siirtojenMaara[x][y] + 1;
                        ensimmainenIollutApu = true;
                    }
                }
            }
            if (ensimmainenIollutApu) {
                ensimmainenIollut = true;
                ensimmainenIollutApu = false;
            }
            // System.out.println("I: " + i);
            for (int kopiointiY = 0; kopiointiY < koko; kopiointiY++) {
                for (int kopiointiX = 0; kopiointiX < koko; kopiointiX++) {
                    siirtojenMaara[kopiointiX][kopiointiY] = (long) Math.round(siirtojenMaara[kopiointiX][kopiointiY] % 1000000007);
                }
            }

        }
        long siirtojayhteensä = 0;
        for (int y = 0; y < koko; y++) {
            for (int x = 0; x < koko; x++) {
                siirtojayhteensä = siirtojayhteensä + siirtojenMaara[x][y];
            }
        }
        System.out.println((int) Math.round(siirtojayhteensä % 1000000007));
    }
}

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