蓝桥杯 ADV-279 算法提高 矩阵乘法

算法提高 矩阵乘法

时间限制：1.0s   内存限制：256.0MB

 

问题描述

　　小明最近刚刚学习了矩阵乘法，但是他计算的速度太慢，于是他希望你能帮他写一个矩阵乘法的运算器。

 

输入格式

　　输入的第一行包含三个正整数N,M,K，表示一个N*M的矩阵乘以一个M*K的矩阵。接下来N行，每行M个整数，表示第一个矩阵。再接下来的M行，每行K个整数，表示第二个矩阵。

 

输出格式

　　输出有N行，每行K个整数，表示矩阵乘法的结果。

 

样例输入

2 1 3
1
2
1 2 3

 

样例输出

1 2 3
2 4 6

 

数据规模和约定

　　0<N,M,K<=100, 0<=矩阵中的每个数<=1000。

 

#include <stdio.h>

int main()
{
    int N, M, K;
    int A[105][105] = { 0 }, B[105][105] = { 0 }, C[105][105] = { 0 };

    scanf("%d %d %d", &N, &M, &K);
    for (int i = 0; i < N; ++i)
        for (int j = 0; j < M; ++j)
            scanf("%d", &A[i][j]);
    for (int i = 0; i < M; ++i)
        for (int j = 0; j < K; ++j)
            scanf("%d", &B[i][j]);

    for (int i = 0; i < N; ++i)
    {
        for (int j = 0; j < K; ++j)
        {
            C[i][j] = 0;
            for (int k = 0; k < M; ++k)
                C[i][j] += A[i][k] * B[k][j];
        }
    }

    for (int i = 0; i < N; ++i)
    {
        for (int j = 0; j < K; ++j)
            printf("%d ", C[i][j]);
        printf("\n");
    }

    return 0;
}