华为机试HJ69：矩阵乘法

作者：Steven
版权声明：著作权归作者所有，商业转载请联系作者获得授权，非商业转载请注明出处

题目描述：

如果A是个x行y列的矩阵，B是个y行z列的矩阵，把A和B相乘，其结果将是另一个x行z列的矩阵C。这个矩阵的每个元素是由下面的公式决定的

矩阵的大小不超过100*100

输入描述：

输入包含多组数据，每组数据包含：
第一行包含一个正整数x，代表第一个矩阵的行数
第二行包含一个正整数y，代表第一个矩阵的列数和第二个矩阵的行数
第三行包含一个正整数z，代表第二个矩阵的列数
之后x行，每行y个整数，代表第一个矩阵的值
之后y行，每行z个整数，代表第二个矩阵的值

输出描述：

对于每组输入数据，输出x行，每行z个整数，代表两个矩阵相乘的结果

示例：

输入：

2
3
2
1 2 3
3 2 1
1 2
2 1
3 3

输出：

14 13
10 11

说明：

1 2 3
3 2 1 
乘以
1 2
2 1
3 3
等于
14 13
10 11 

解题思路：

本题是通过数组实现矩阵计算。按照给定的行列分别定义好A和B两个矩阵的尺寸，并完成赋值；用过calcMatrix函数实现矩阵相乘，每个位置的值是A对应行和B对应列进行点乘的结果，完毕。

测试代码：

#include <iostream>
#include <vector>

using namespace std;

vector<vector<int>> calcMatrix(vector<vector<int>> A,vector<vector<int>> B,int row,int col)
{
    vector<vector<int>> R(row,vector<int>(col,0));
    for(int i=0;i<row;++i)
    {
        for(int j=0;j<col;++j)
        {
            int sum=0;
            for(int m=0,n=0;m<A[i].size(),n<B.size();++m,++n)
            {
                sum+=A[i][m]*B[n][j];
            }
            R[i][j]=sum;
        }
    }
    return R;
}

int main()
{
    int row1,col1,row2,col2;
    while(cin>>row1>>col1>>col2)
    {
        row2=col1;
        vector<vector<int>> A(row1,vector<int>(col1,0));
        vector<vector<int>> B(row2,vector<int>(col2,0));
        for(int i=0;i<row1;++i)
        {
            for(int j=0;j<col1;++j)
            {
                cin>>A[i][j];
            }
        }
        for(int i=0;i<row2;++i)
        {
            for(int j=0;j<col2;++j)
            {
                cin>>B[i][j];
            }
        }
        vector<vector<int>> result=calcMatrix(A,B,row1,col2);
        for(int i=0;i<row1;++i)
        {
            for(int j=0;j<col2;++j)
            {
                cout<<result[i][j]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}

标签：row1,int,col2,矩阵,col1,华为,vector,机试,HJ69
来源： https://blog.csdn.net/zhaitianbao/article/details/120257159