#include <stdio.h>
#include <stdlib.h>
#define MAX 4

float prosjek(int matrica[]);

int main() {
    int matrica[MAX][MAX];
    int i, j;
    float avg;

    printf("Unesite %d elemenata:\n", MAX*MAX);
    for (i = 0; i<MAX; i++)
    for (j = 0; j<MAX; j++){
        printf("Unesite element [%d][%d] = ", i, j);
        scanf_s("%d", &matrica[i][j]);
    }

    for (i = 0; i<MAX; i++) {
        for (j = 0; j<MAX; j++)
            printf("%d\t", matrica[i][j]);
        printf("\n");
    }

    avg = prosjek(&matrica[0][0]);

    printf("\n\nProsjek elemenata: %.2f", avg);

    return 0;
}

float prosjek(int matrica[]) {
    float *average, vrati, suma;
    int i, j;

    average = (float*)malloc(MAX*sizeof(float));

    for (i = 0; i<MAX; i++) {
        suma = 0;
        for (j = 0; j<MAX; j++) {
            suma += matrica[j*MAX + i];
        }
        *(average + i) = suma / MAX;
    }

    printf("\nIspis polja:\n");
    for (i = 0; i<MAX; i++)
        printf("%.2f\t", *(average + i));

    suma = 0;

    for (i = 0; i<MAX; i++)
        suma += *(average + i);

    vrati = suma / MAX;

    free(average);

    return vrati;
}