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

矩阵乘法算法优化

最新推荐文章于 2023-01-09 18:31:10 发布

原创最新推荐文章于 2023-01-09 18:31:10 发布 · 513 阅读

1 ·

CC 4.0 BY-SA版权

蓝桥杯专栏收录该内容

232 篇文章

订阅专栏

本文介绍了一种矩阵乘法的算法实现，通过三维循环结构，有效地完成了两个矩阵的相乘运算。该算法适用于N*M矩阵与M*K矩阵的乘法，能够处理的数据规模为0<N,M,K<=100，矩阵元素范围在0到1000之间。文章提供了完整的C语言代码实现，包括输入矩阵、计算结果和输出结果的全过程。

算法提高矩阵乘法

时间限制：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;
}