2015.网研院.Problem B.矩阵连乘-蒲公英云

2015.网研院.Problem B.矩阵连乘

题目描述
给你3个5*5的矩阵A、B、C，让你求他们依次相乘的结果矩阵D,D=ABC，

输入描述
有多组数据，第一行是数据总数。
每组数据输入三个5*5的矩阵，每个数的取值范围是[0，10]

输出描述
输出一个矩阵D即答案
输入：

输出：

886 734 804 1112 832
2074 1700 1852 2634 2005
1791 1462 1589 2280 1740
942 722 770 1234 954
848 662 710 1072 788
#include<bits/stdc++.h>
using namespace std;
int main() { 
    int T,sum;
    int a[5][5],b[5][5],c[5][5];
    int tmp[5][5],ans[5][5];
    cin >> T;
    while(T--) { 
        for(int i=0; i<5; i++)
            for(int j=0; j<5; j++) { 
                cin>>a[i][j];
            }
        for(int i=0; i<5; i++)
            for(int j=0; j<5; j++) { 
                cin >>b[i][j];
            }
        for(int i=0; i<5; i++)
            for(int j=0; j<5; j++) { 
                cin >>c[i][j];
            }
        for(int i=0; i<5; i++) { 
            for(int j=0; j<5; j++) { 
                sum=0;
                for(int k=0; k<5; k++) { 
                    sum+=a[i][k]*b[k][j];
                }
                tmp[i][j]=sum;
            }
        }
        for(int i=0; i<5; i++) { 
            for(int j=0; j<5; j++) { 
                sum=0;
                for(int k=0; k<5; k++) { 
                    sum+=tmp[i][k]*c[k][j];
                }
                ans[i][j]=sum;
            }
        }
        for(int i=0; i<5; i++) { 
            int flag=0;
            for(int j=0; j<5; j++) { 
                if(flag==0){ 
                    cout<<ans[i][j];
                    flag=1;
                }
                else{ 
                    cout<<" "<<ans[i][j];
                }
            }
            cout<<endl;
        }
    }
}