csp202305-2 矩阵运算

最新推荐文章于 2024-03-04 19:00:59 发布

葵花zi

最新推荐文章于 2024-03-04 19:00:59 发布

阅读量756

点赞数

分类专栏： cspccf 文章标签：矩阵 c++ 算法

本文链接：https://blog.csdn.net/l141930402/article/details/131293492

版权

cspccf 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

文章介绍了一种优化矩阵运算的方法，通过先执行k*v（d*d）而非先执行q*k（n*n），在d远小于n的情况下减少计算量和内存使用。程序实现了这一优化策略，对输入的矩阵q,k,v和w进行相应运算，输出结果矩阵。

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

优化方法：先k*v，再q*k。（先q*k：n*d×d*n=n*n，改为先k*v：d*n×n*d=d*d，此题d远比n小，所以优化可行）。

#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
const ll maxn = 10005;
ll q[maxn][25];
ll k[25][maxn];
ll v[maxn][25];
ll w[maxn];
ll n, d;
ll kv[25][25];
ll ww[maxn][25];
int main()
{
    cin >> n >> d;
    for (ll i = 1; i <= n; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            cin >> q[i][j];
        }
    }
    for (ll i = 1; i <= n; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            cin >> k[j][i];//k转置
        }
    }
    for (ll i = 1; i <= n; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            cin >> v[i][j];
        }
    }
    for (ll i = 1; i <= n; i++)
    {
        cin >> w[i];
    }
    // k*v
    ll ans = 0;
    for (ll i = 1; i <= d; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            for (ll kk = 1; kk <= n; kk++)
            {
                ans += k[i][kk] * v[kk][j];
            }
            kv[i][j] = ans;
            ans = 0;
        }
    }
    
    for (ll i = 1; i <= n; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            for (ll kk = 1; kk <= d; kk++)
            {
                ans += q[i][kk] * kv[kk][j];
            }
            ww[i][j] = ans;
            ans = 0;
        }
    }
    for (ll i = 1; i <= n; i++)
    {
        for (ll j = 1; j <= d; j++)
        {
            ww[i][j] *= w[i];
            cout << ww[i][j] << " ";
        }
        cout << "\n";
    }
}