51nod 1137 矩阵乘法
给出2个N * N的矩阵M1和M2,输出2个矩阵相乘后的结果。
Input
第1行:1个数N,表示矩阵的大小(2 <= N <= 100) 第2 - N + 1行,每行N个数,对应M1的1行(0 <= M1[i] <= 1000) 第N + 2 - 2N + 1行,每行N个数,对应M2的1行(0 <= M2[i] <= 1000)
Output
输出共N行,每行N个数,对应M1 * M2的结果的一行。
Input示例
2 1 0 0 1 0 1 1 0
Output示例
0 1 1 0
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <queue>
#include <stack>
#include <vector>
using namespace std;
#define INF 0x3f3f3f
#define pi acos(-1.0)
#define MAX 1000010
#define N 105
struct Mat
{
int p[N][N];
};
int n;
Mat mul(Mat a,Mat b)
{
Mat c;
memset(c.p,0,sizeof(c.p));
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
for(int k = 0; k < n; k++)
c.p[i][j] += a.p[i][k]*b.p[k][j];
return c;
}
int main()
{
scanf("%d",&n);
Mat A,B;
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
scanf("%d",&A.p[i][j]);
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
scanf("%d",&B.p[i][j]);
Mat C = mul(A,B);
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
printf("%d%c",C.p[i][j],j==n-1?'\n':' ');
}