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

Love The Way You Lie 2023-07-17 15:44 79阅读 0赞

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

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

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

1
    1 2 3 4 5
    6 5 8 9 7
    6 5 7 8 4
    5 6 3 2 1
    2 3 1 4 5
    1 2 3 4 5
    3 2 1 4 5
    6 5 4 1 2
    6 2 1 4 5
    3 2 1 3 4
    7 8 9 6 3
    1 2 4 5 6
    3 2 1 4 5
    3 2 1 8 6
    3 0 0 1 0

**输出：**

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;
    		}
    	}
    }