Illuminati

#include <bits/stdc++.h>
using namespace std;

#define int long long
int n;

void multiply(int** A, int** B, int** C)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            C[i][j] = 0;
            for (int k = 0; k < n; k++)
                C[i][j] += A[i][k] * B[k][j];
        }
    }
}

int getTrace(int** graph)
{
    int trace = 0;
    for (int i = 0; i < n; i++)
        trace += graph[i][i];
    return trace;
}

int triangleInGraph(int** graph)
{
    // Allocate memory for aux2 and aux3 matrices
    int** aux2 = new int*[n];
    int** aux3 = new int*[n];
    for (int i = 0; i < n; i++)
    {
        aux2[i] = new int[n]();
        aux3[i] = new int[n]();
    }

    // Perform matrix multiplications
    multiply(graph, graph, aux2);
    multiply(graph, aux2, aux3);

    // Get trace of aux3
    int trace = getTrace(aux3);

    // Deallocate memory
    for (int i = 0; i < n; i++)
    {
        delete[] aux2[i];
        delete[] aux3[i];
    }
    delete[] aux2;
    delete[] aux3;

    return trace / 6;
}

signed main()
{
    cin >> n;

    // Dynamically allocate the graph matrix
    int** graph = new int*[n];
    for (int i = 0; i < n; i++)
        graph[i] = new int[n];

    // Read the input matrix
    for (int i = 0; i < n; i++) {
        string temp;
        cin >> temp;
        for (int j = 0; j < n; j++) {
            graph[i][j] = temp[j] - '0';
        }
    }

    cout << triangleInGraph(graph) << endl;

    // Deallocate memory
    for (int i = 0; i < n; i++)
        delete[] graph[i];
    delete[] graph;

    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