CSP CCF认证202305-2 矩阵运算

最新推荐文章于 2024-02-26 16:59:37 发布

听风雨233

最新推荐文章于 2024-02-26 16:59:37 发布

阅读量278

点赞数 2

文章标签：算法 c++

本文链接：https://blog.csdn.net/m0_62560273/article/details/132940272

版权

文章介绍了如何通过改变矩阵乘法的顺序来优化计算过程，尤其是在处理大规模数组和数据范围时，以减少内存使用并提高效率。C++代码展示了如何先计算矩阵K的转置与矩阵V的乘法，再与矩阵Q进行乘积运算，并考虑了INT范围限制。

摘要由CSDN通过智能技术生成

思路：

通过改变矩阵乘法顺序优化计算
合理评估数组大小与数据范围

代码：

#include<iostream>
using namespace std;
const int N=1e4+3;
const int D=25;
typedef long long ll;
int q[N][D],k[N][D],v[N][D];
ll ans[N][D],t[N][D]; //两步中间结果可能会超出INT范围
int w[N];
int main()
{
    int n,d;
    scanf("%d%d",&n,&d);
    for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
    {
        scanf("%d",&q[i][j]);
    }
        for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
    {
        scanf("%d",&k[i][j]);
    }
        for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
    {
        scanf("%d",&v[i][j]);
    }
    for(int i=0;i<n;i++)
    {
        scanf("%d",&w[i]);
    }
    //先计算矩阵K的转置与矩阵V的乘法
    for(int c=0;c<d;c++)
    for(int i=0;i<d;i++)
    {
        ll mid = 0;
        for(int j=0;j<n;j++)
        {
             mid += k[j][c] * v[j][i];
        }
        t[c][i]=mid;
    }
    //计算Q与上一个计算所得矩阵的乘积，同时乘以W相量
      for(int c=0;c<n;c++)
    for(int i=0;i<d;i++)
    {
        ll mid = 0;
        for(int j=0;j<d;j++)
        {
             mid += q[c][j] * t[j][i];
        }
        ans[c][i]=mid;
        ans[c][i]*=w[c];

    }

    for(int i=0;i<n;i++)
    {
        for(int j=0;j<d;j++)
        {
            cout<<ans[i][j]<<" ";
        }
        puts("");
    }

}