算法訓練 矩陣乘法 （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;
}