华为机试HJ69:矩阵乘法

简介: 华为机试HJ69:矩阵乘法

题目描述:

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


相关文章
|
算法
华为机试HJ14:字符串排序
华为机试HJ14:字符串排序
|
人工智能
华为机试HJ26:字符串排序
华为机试HJ26:字符串排序
华为机试HJ107:求解立方根
华为机试HJ107:求解立方根
153 1
|
算法
华为机试HJ108:求最小公倍数
华为机试HJ108:求最小公倍数
110 1
|
容器
华为机试HJ102:字符统计
华为机试HJ102:字符统计
171 1
|
容器
华为机试HJ89:24点运算
华为机试HJ89:24点运算
150 0
|
算法
华为机试HJ82:将真分数分解为埃及分数
华为机试HJ82:将真分数分解为埃及分数
108 0
华为机试HJ75:公共子串计算
华为机试HJ75:公共子串计算
101 0

热门文章

最新文章