Illuminati

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 3005;

int a[N][N], b[N][N], c[N][N];
int n;

void Calc()
{
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            b[i][j] = 0;
            for(int k=1; k<=n; k++)
                b[i][j] += a[i][k] * c[k][j];
        }
    }
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            c[i][j] = b[i][j];
}


void Calc2()
{
    for(int i=1; i<=n; i++)
    {
        b[i][i] = 0;
        for(int k=1; k<=n; k++)
            b[i][i] += a[i][k] * c[k][i];
    }
    for(int i=1; i<=n; i++)
        c[i][i] = b[i][i];
}

int main()
{
    cin >> n;
    string str;
    for(int i=1; i<=n; i++)
    {
        cin >> str;
        for(int j=1; j<=n; j++)
        {
            a[i][j] = (str[j-1]=='1'? 1:0);
            c[i][j] = a[i][j];
        }
    }

    Calc();
    Calc2();
    long long ans = 0;
    for(int i=1; i<=n; i++)
        ans += c[i][i];
    cout << ans/6;
    
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: ACCEPTED

Test 28

Verdict: ACCEPTED

Test 29

Verdict: ACCEPTED

Test 30

Verdict: ACCEPTED

Test 31

Verdict: ACCEPTED

Test 32

Verdict: ACCEPTED

Test 33

Verdict: ACCEPTED

Test 34

Verdict: ACCEPTED

Test 35

Verdict: ACCEPTED

Test 36

Verdict: ACCEPTED

Test 37

Verdict: ACCEPTED

Test 38

Verdict: ACCEPTED

Test 39

Verdict: ACCEPTED

Test 40

Verdict: ACCEPTED

Test 41

Verdict: ACCEPTED

Test 42

Verdict: ACCEPTED

Test 43

Verdict: ACCEPTED

Test 44

Verdict: ACCEPTED

Test 45

Verdict: ACCEPTED

Test 46

Verdict: ACCEPTED

Test 47

Verdict: ACCEPTED

Test 48

Verdict: TIME LIMIT EXCEEDED

Test 49

Verdict: TIME LIMIT EXCEEDED

Test 50

Verdict: TIME LIMIT EXCEEDED

Test 51

Verdict: TIME LIMIT EXCEEDED

Test 52

Verdict: TIME LIMIT EXCEEDED

Test 53

Verdict: TIME LIMIT EXCEEDED