CCF-CSP真题《202305-2 矩阵运算》

代码

#include<stdio.h>
#include<stdlib.h>
#define maxsize 1000

int main()
{
    int n,d;
    scanf("%d %d",&n,&d);
    int a[3][n][d];
    int w[n];
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<n;j++)
            for(int k=0;k<d;k++)
                scanf("%d",&a[i][j][k]);
    }
    for(int i = 0; i<n; i++)
        scanf("%d",&w[i]);
    //计算
    //转置，可省略
    //矩阵乘法
    //矩阵点乘
    //矩阵乘法
    //改变计算顺序可减小时间复杂度
    //K*V=tem1
    int tem1[d][d];
    for(int i=0;i<d;i++)
        for(int j=0;j<d;j++)
            tem1[i][j]=0;

    for(int i=0;i<d;i++)
        for(int j=0;j<d;j++)
            for(int k=0;k<n;k++)
                tem1[i][j]+=a[1][k][i]*a[2][k][j];
    //tem2=Q*tem1
    int tem2[n][d];
    for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
            tem2[i][j]=0;
    for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
            for(int k=0;k<d;k++)
                tem2[i][j]+=a[0][i][k]*tem1[k][j];
    //ans=W*tem2
    int ans[n][d];
    for(int i=0;i<n;i++)
        for(int j=0;j<d;j++)
            ans[i][j]=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<d;j++)
        {
            ans[i][j]=w[i]*tem2[i][j];
            printf("%d ",ans[i][j]);
        }
        printf("\n");
    }

    return 0;
}

结果

 c++可以

CCF-CSP真题《202305-2 矩阵运算》思路+python，c++满分题解_ccf csp题解_Hulake_的博客-CSDN博客