#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
int main()
{
int n,d;//n行d列
cin>>n>>d;
int Q[1001][21],K[1001][21],V[1001][21];
int W[n];
for(int i=1;i<=3;i++)//输入三个矩阵
{
for(int j=0;j<n;j++)//输入每个矩阵的行
{
for(int t=0;t<d;t++)//输入矩阵的一行的每个数
{
if(i==1)
{
cin>>Q[j][t];
}
else
{
if(i==2) cin>>K[j][t];
else
if(i==3)
cin>>V[j][t];
}
}
}
}
for(int l=0;l<n;l++)
{
cin>>W[l];
}
int KT[21][1001];//K的转置矩阵
for(int i=0;i<n;i++)//转置矩阵KT的列
{
for(int j=0;j<d;j++)//转置矩阵KT的行
{
KT[j][i]=K[i][j];
}
}
//Q*KT
int QKT[1001][1001];//要具体的大小,尽量不要用QKT[n][n];否则会出现错误 , 大小不够的问题
for(int i=0;i<n;i++)//Q的行
{
for(int l=0;l<n;l++)//KT的列
{
for(int j=0;j<d;j++)
{
QKT[i][l]+=(Q[i][j]*KT[j][l]);
}
}
}
//W*QKT
for(int h=0;h<n;h++)
{
for(int l=0;l<n;l++)
{
QKT[h][l]*=W[h];
}
}
int Final[1001][21];//WQKT*V----n*n n*d
for(int h=0;h<n;h++)
{
for(int l=0;l<d;l++)
{
for(int x=0;x<n;x++)
{
Final[h][l]+=QKT[h][x]*V[x][l];
}
}
}
for(int i=0;i<n;i++)
{
for(int j=0;j<d;j++)
{
if(j!=d-1)
cout<<Final[i][j]<<" ";
else cout<<Final[i][j];
}
if(i!=n-1)
cout<<endl;
}
return 0;
}
没有满分,只有七十分
注意的点:二维数组的大小尽量用明确的数字,不要用输入的变量来定义
优化:先算KT*V再往前算,因为从前往后算会出现n*n的矩阵,从后往前算不会,可以避免超时
涉及矩阵运算,记住矩阵乘法公式,看一下怎样的顺序运算可以避免出现某些比较大的矩阵从而节约时间
#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
int Q[10001][21],K[10001][21],V[10001][21];
int KT[21][10001];//K的转置矩阵
long long KTV[10001][21];
long long QKT[10001][21];//要具体的大小,尽量不要用QKT[n][n];否则会出现错误 , 大小不够的问题
int main()
{
int n,d;//n行d列
cin>>n>>d;
int W[n];
for(int i=1;i<=3;i++)//输入三个矩阵
{
for(int j=0;j<n;j++)//输入每个矩阵的行
{
for(int t=0;t<d;t++)//输入矩阵的一行的每个数
{
if(i==1)
{
cin>>Q[j][t];
}
else
{
if(i==2) cin>>K[j][t];
else
if(i==3)
cin>>V[j][t];
}
}
}
}
for(int l=0;l<n;l++)
{
cin>>W[l];
}
for(int i=0;i<n;i++)//转置矩阵KT的列
{
for(int j=0;j<d;j++)//转置矩阵KT的行
{
KT[j][i]=K[i][j];
}
}
//KT*V d n * n d
for(int i=0;i<d;i++)//KT 的行
{
for(int l=0;l<d;l++)//V的列和结果的列
{
for(int j=0;j<n;j++)
{
KTV[i][l]+=KT[i][j]*V[j][l];
}
}
}
//Q*KTV n d * d d
for(int i=0;i<n;i++)//Q的行
{
for(int l=0;l<d;l++)//KT的列
{
for(int j=0;j<d;j++)
{
QKT[i][l]+=(Q[i][j]*KTV[j][l]);
}
QKT[i][l]*=(long long)W[i];
}
}
for(int i=0;i<n;i++)
{
for(int j=0;j<d;j++)
{
if(j!=d-1)
cout<<QKT[i][j]<<" ";
else cout<<QKT[i][j];
}
if(i!=n-1)
cout<<endl;
}
return 0;
}