算法训练 矩阵乘法 （vector使用案例）

问题描述

　　输入两个矩阵，分别是m*s，s*n大小。输出两个矩阵相乘的结果。

输入格式

　　第一行，空格隔开的三个正整数m,s,n（均不超过200）。
　　接下来m行，每行s个空格隔开的整数，表示矩阵A（i，j）。
　　接下来s行，每行n个空格隔开的整数，表示矩阵B（i，j）。

输出格式

　　m行，每行n个空格隔开的整数，输出相乘後的矩阵C（i，j）的值。

样例输入

2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1

样例输出

-3 2
-8 2

提示
矩阵C应该是m行n列，其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1)*(0,1,3) = 1 * 0 +0*1+(-1)*3=-3

 

思路：本题我采用动态数组vetor来存储数据，也可以直接开三个数组，使用三重循环来解决问题。为了熟悉vector的使用，我在这里使用了vector，具体代码如下：

///矩阵乘法
#include <iostream>
#include <vector>
using namespace std;
int a[1000][1000];
int b[1000][1000];
int main()
{
    vector<int> v[1000];
    vector<int> vv[1000];
    int m,s,n;
    cin>>m>>s>>n;
    for(int i=0;i<m;i++)
    {
        for(int j=0;j<s;j++)
        {
            cin>>a[i][j];
            v[i].push_back(a[i][j]);
        }
    }
    for(int i=0;i<s;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>b[i][j];
            vv[j].push_back(b[i][j]);
        }
    }

    for(int i=0;i<m;i++)
    {
        for(int j=0;j<n;j++)
        {
            int ans=0;
            for(int k=0;k<s;k++)
            {
                ans+=v[i][k]*vv[j][k];
            }
            cout<<ans<<" ";
        }
        cout<<endl;
    }
    return 0;
}