csp矩阵运算

 

 

两个点：

做公式变换+long long int 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;

int n,d;
long long int Q[10010][25],K[10010][25],V[10010][25];
long long int K_T[25][10010];
long long int W[10010];
long long int K_T_V[25][25];
long long int Q_K_T_V[10010][25];


int main(){
    scanf("%d%d",&n,&d);
    //输入Q
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            scanf("%lld",&Q[i][j]);
        }
    }
    //输入K--n*d
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            scanf("%lld",&K[i][j]);
            K_T[j][i]=K[i][j];//计算K的转置--d*n
        }
    }
    //输入V
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            scanf("%lld",&V[i][j]);
        }
    }
    //输入W
    for(int i=1;i<=n;i++){
        scanf("%lld",&W[i]);
    }

    //先算K_T*V
    for(int i=1;i<=d;i++){
        for(int j=1;j<=d;j++){
            long long int res=0;
            for(int k=1;k<=n;k++){
                res+=K_T[i][k]*V[k][j];
            }
            K_T_V[i][j]=res;
        }
    }

    //再算Q*(K_T*V)
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            long long int res=0;
            for(int k=1;k<=d;k++){
                res+=Q[i][k]*K_T_V[k][j];
            }
            Q_K_T_V[i][j]=res;
        }
    }

    
    //进行点乘
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            Q_K_T_V[i][j]*=W[i];
        }
    }
    
    
    for(int i=1;i<=n;i++){
        for(int j=1;j<=d;j++){
            printf("%lld ",Q_K_T_V[i][j]);
        }
        printf("\n");
    }
    return 0;
    
    
}