华为oj矩阵相乘2

#include "oj.h"
#include <stdio.h>

int main()
{
int i=0,j=0;
int A[3][3] = {
{1, 2, 3},
{4, 5, 6},
{7, 8, 9}};
int B[3][3] = {
{1, 2, 3},
{4, 5, 6},
{7, 8, 9}};
int C[3][3] = {0};

matrix((int **)A, (int **)B, (int **)C, 3);
for(i=0;i<3;i++)
{
for(j=0;j<3;j++)
{
printf("%d ",C[i][j]);
}
printf("\n");
}

return 0;
}

/*
功能: 矩阵相乘

输入: MatrixA,MatrixB

输出: MatrixC

返回: 0

*/

int matrix(int **MatrixA, int **MatrixB, int **MatrixC, int N)
{
int i, j, k;
int a, b;
if(MatrixA == NULL || MatrixB == NULL || MatrixC == NULL)
{
return -1;
}

for(i=0;i<N;i++)
{
for(j=0;j<N;j++)
{
*((int *)MatrixC + i*N +j) = 0;	//转化成一维数组,MatrixC[i][j]
for(k=0;k<N;k++)
{
a = *((int *)MatrixA + i*N + k); //MatrixA[i][k],对于MatrixA，行i是确定的
b = *((int *)MatrixB + k*N + j); //MatrixB[k][j],对于MatrixB，列j是确定的
*((int *)MatrixC + i*N + j) += a*b;
}
}
}

return 0;
}