华为机试HJ69:矩阵乘法

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

题目描述:

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

华为机试HJ69:矩阵乘法

矩阵的大小不超过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;
}

上一篇:hive基础


下一篇:Pandas-排序函数-sort_values()