csp202305-2矩阵运算

1. 矩阵：之前矩阵的知识点有点忘记了，又稍微复习了一下。

关于矩阵乘法，A行数等于B列数时乘法才有意义。

对于矩阵运算，改变顺序是一个很常用的优化方法。

Anm*Bmk的代码需要三层循环，时间复杂度达到Onmk。所以我们希望m尽可能小。

2. 看了下矩阵求逆，好麻烦，下次遇到了再学吧哈哈。 

#include<bits/stdc++.h> 
using namespace std;

int main() {  
    long long n, d;
    cin >> n >> d;

    vector<vector<long long>>Q(n, vector<long long>(d));
    vector<vector<long long>>K_T(d, vector<long long>(n));
    vector<vector<long long>>V(n, vector<long long>(d));
    vector<long long>W(n);

    // 输入Q
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < d; j++)
        {
            cin >> Q[i][j];
        }
    }
    // 输入K_T
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < d; j++)
        {
            cin >> K_T[j][i];
        }
    }
    // 输入V
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < d; j++)
        {
            cin >> V[i][j];
        }
    }
    // 输入W
    for (int i = 0; i < n; i++)
    {
        cin >> W[i];
    }

    // 计算 W * Q
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < d; j++)
        {
            Q[i][j] *= W[i];
        }
    }
     
    // 计算 K_T * V
    vector<vector<long long>>T1(d, vector<long long>(d));
    for (int i = 0; i < d; i++)
    {
        for (int j = 0; j < d; j++)
        {
            for (int k = 0; k < n; k++)
            {
                T1[i][j] += K_T[i][k] * V[k][j];
            }
        }
    }

    // 计算 Q * T1
    vector<vector<long long>>T2(n, vector<long long>(d));
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < d; j++)
        {
            for (int k = 0; k < d; k++)
            {
                T2[i][j] += Q[i][k] * T1[k][j];
            }
        }
    }

    for (const auto& it : T2) {
        for (const auto& jt : it) {
            cout << jt << " ";
        }
        cout << endl;
    }

    return 0;
}