C Program to Multiply Two Matrix Using Multi-dimensional Arrays

#include <stdio.h>
#include <stdlib.h>
int ** matrixMult(int **a, int **b, int r, int c) {
    int i, j, k, sum = 0;
    int **result = malloc(r * sizeof (int *));
    for (i = 0; i < r; i++) {
        result[i] = malloc(r * sizeof (int));
    }
    for (i = 0; i < r; i++) {
        for (j = 0; j < r; j++) {
            for (k = 0; k < c; k++) {
                sum += a[i][k] * b[k][j];
            }
            result[i][j] = sum;
            sum = 0;
        }
    }
    return result;
}
int main() {
    int i, j, rows, cols, **a, **b, **result;
    printf("Multiplication of two Matrix");
    printf("\nEnter number of rows :");
    scanf("%d", &rows);
    printf("\nEnter number of columns :");
    scanf("%d", &cols);
    // allocate rows, each row is a pointer to int
    a = malloc(rows * sizeof (int *));
    // allocate cols(rows), each row is a pointer to int
    b = malloc(cols * sizeof (int *));
    // for each row allocate cols int
    for (i = 0; i < rows; i++) {
        a[i] = malloc(cols * sizeof (int));
    }
    for (i = 0; i < cols; i++) {
        b[i] = malloc(rows * sizeof (int));
    }
    printf("\nEnter Matrix A\n");
    for (i = 0; i < rows; i++) {
        for (j = 0; j < cols; j++) {
            printf("Enter Element %d %d : ", i, j);
            scanf("%d", &a[i][j]);
        }
    }
    printf("\nEnter Matrix B\n");
    for (i = 0; i < cols; i++) {
        for (j = 0; j < rows; j++) {
            printf("Enter Element %d %d : ", i, j);
            scanf("%d", &b[i][j]);
        }
    }
    result = matrixMult(a, b, rows, cols);
    printf("\nMultiplication results of Matrix A & B\n");
    for (i = 0; i < rows; i++) {
        for (j = 0; j < rows; j++) {
            printf("%d ", result[i][j]);
        }
        printf("\n");
    }
    return 0;
}