矩阵相乘（自用）

矩阵相乘（自用）

离散数学矩阵自乘算的太麻烦了（bushi）
输入俩矩阵的行列（因为矩阵一的列数和矩阵二的行数相等，所以n一起表示这俩）
（如果方阵自乘，直接给b复制a）

#include<iostream>
using namespace std;
#define ll long long
ll a[1000][1000];
ll b[1000][1000];
ll ans[1000][1000];
int main(){
    while(true){
        ll m,n,q;
        cin>>m>>n>>q;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        memset(ans,0,sizeof(ans));
        for(ll i=1;i<=m;i++){
            for(ll j=1;j<=n;j++){
                cin>>a[i][j];
                if(m==n&&n==q){
                    b[i][j]=a[i][j];
                }
            }
        }
        if(m!=n||n!=q){
            for(ll i=1;i<=n;i++){
                for(ll j=1;j<=q;j++){
                cin>>b[i][j];
                }
            }
        }
        for(ll i=1;i<=m;i++){
            for(ll j=1;j<=n;j++){
                for(ll k=1;k<=q;k++){
                    ans[i][k]+=a[i][j]*b[j][k];
                }
            }
        }
        for(ll i=1;i<=m;i++){
            for(ll j=1;j<=q;j++){
                printf("%5lld",ans[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}