#include<stdio.h>
#include<stdlib.h>
#define NUM_ROWS 8

typedef int (*init_fn_t)(int rows, int cols, int dim); 

int A_value(int i, int j, int n)
{
   return 1 + (i + j) % n;
}

int Result_value(int i, int j, int n)
{
   return n * (n + 1)/ 2;
}

int B_value(int i, int j, int n)
{
   return 1;
}


void init_matrix(int **A, init_fn_t f, int n) {
	int i,j;
	for(i=0;i<n;++i){
		for(j=0;j<n;++j)
			A[i][j] = f(i,j,n);
	}
}

#define max(i,j) (i > j ? i : j)
#define min(i,j) (i < j ? i : j)
int A_value2(int i,int j, int n)
{
  return max(0, j-i+1);
}

int B_value2(int i, int j, int n)
{
  return max(0, i-j+1);
}

int Result_value2(int i, int j, int n)
{
	 int ni = n - i;
   if (i == j) return (ni * (ni + 1)*(2*ni+1))/6; 
   int l = max(i,j);
   int s = min(i,j);
   int ns = n - l;
	 int sigma_n2 = (ns*(ns+1)*(2*ns+1))/6;
	 int sigma_n =  (ns*(ns+1))/2;
	 return sigma_n2 + (l-s)*sigma_n; 	
}

void print_matrix(int **A) {
	int i,j;
	printf("\n Printing the result \n");
	for(i=0;i<NUM_ROWS;++i){
		for(j=0;j<NUM_ROWS;++j)
			printf("%d\t",A[i][j]);
		printf("\n");
	}
	printf("\n End of printing the result \n");
}

void do_matrix_multiply(int **A, int **B, int **result, int n) {
	int i,j,k;	

	for(i=0;i<n;++i) {
		for(j=0;j<n;++j) {
			int t = 0;
			for(k=0;k<n;++k) t += (A[i][k]*B[k][j]);
			if(t !=result[i][j]) printf("\n problem with %d,%d\n",i,j);
    }
  }
}


int **alloc_matrix(n)
{
  int i;
  int **A = (int **)malloc(sizeof(int *)*n);
	for(i=0;i<n;++i) A[i] = (int *)malloc(sizeof(int)*n);
  return A;
}


int main(int argc, char *argv[]) {
	int i,j;
  
  int n = NUM_ROWS;

  int **A = alloc_matrix(n);
  int **B = alloc_matrix(n);
  int **result = alloc_matrix(n);

//TEST1 starts here
  init_matrix(A, A_value, n);
  init_matrix(B, B_value, n);
  init_matrix(result, Result_value, n);

	print_matrix(A);
	print_matrix(B);
	print_matrix(result);
	do_matrix_multiply(A, B, result, n);

//TEST2 starts here
  init_matrix(A, A_value2, n);
  init_matrix(B, B_value2, n);
  init_matrix(result, Result_value2, n);

	print_matrix(A);
	print_matrix(B);
	print_matrix(result);
	do_matrix_multiply(A, B, result, n);
}