#include <iostream>
#include <cstdlib>
#include <math.h>
#include <omp.h>

using namespace std;

int main(int argc, char *argv[]) {

    if(argc!= 3){
        cout << "Nieprawidlowa ilosc parametrow wejsciowych!";
        return 1;
    }

    int n = atoi(argv[1]);
    int size = atoi(argv[2]);

    int **A,**B,**C,suma;

    A = new int * [size];
    B = new int * [size];
    C = new int * [size];

    for(int i = 0; i < size; i++){
        A[i] = new int [size];
        B[i] = new int [size];
        C[i] = new int [size];
    }

    for(int i = 0; i < size; i++){
        for(int j = 0; j < size; j++){
            A[i][j] = (int)(sin(i) * i * j) % 10;
            B[i][j] = (int)(cos(j) * (i + j)) % 10;
        }
    }

    double ctime1 = omp_get_wtime();
    #pragma omp parallel for default(shared) num_threads(n)
    for(int i = 0; i < size; i++){
        for(int j = 0; j < size; j++){
            suma = 0;
            for(int k = 0; k < size; k++){
                suma += A[i][k] * B[k][j];
                C[i][j] = suma;
            }
        }
    }
    double ctime2 = omp_get_wtime();

    for(int i = 0; i < size; i++){
        delete [] A[i];
        delete [] B[i];
        delete [] C[i];
    }

    delete [] A;
    delete [] B;
    delete [] C;

    cout << "Czas: " << (ctime2-ctime1) <<  "ms";

    return 0;
}