1. 程式人生 > >演算法-藍橋杯-演算法提高 矩陣相乘 (C++)

演算法-藍橋杯-演算法提高 矩陣相乘 (C++)

1 引言

    矩陣乘法,以前做過。

2 題目

問題描述

  小明最近在為線性代數而頭疼,線性代數確實很抽象(也很無聊),可惜他的老師正在講這矩陣乘法這一段內容。
  當然,小明上課打瞌睡也沒問題,但線性代數的習題可是很可怕的。
  小明希望你來幫他完成這個任務。

  現在給你一個ai行aj列的矩陣和一個bi行bj列的矩陣,
  要你求出他們相乘的積(當然也是矩陣)。
  (輸入資料保證aj=bi,不需要判斷)輸入格式  輸入檔案共有ai+bi+2行,並且輸入的所有數為整數(long long範圍內)。
  第1行:ai 和 aj
  第2~ai+2行:矩陣a的所有元素
  第ai+3行:bi 和 bj
  第ai+3~ai+bi+3行:矩陣b的所有元素輸出格式  輸出矩陣a和矩陣b的積(矩陣c)
  (ai行bj列)樣例輸入2 2
12 23
45 56
2 2
78 89
45 56樣例輸出1971 2356
6030 7141

3 原始碼

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
#include <stdlib.h>
#include <map>
using namespace std;
long long a[105][105];//矩陣1
long long b[105][105];//矩陣2
long long c[105][105];//結果矩陣
void in(long long n,long long m,long long (&x)[105][105])//函式呼叫陣列輸入
{
    for(long long i=0;i<n;i++)
        for(long long j=0;j<m;j++)
        cin>>x[i][j];
}
void sout(long long n,long long m,long long l)//矩陣相乘並輸出
{
    memset(c,0,sizeof(c));

    for(long long i=0;i<n;i++)
    {
        for(long long k=0;k<l;k++)
        for(long long j=0;j<m;j++)
        {
            c[i][k]+=a[i][j]*b[j][k];
        }
    }

    for(long long i=0;i<n;i++)
    {
        for(long long j=0;j<l;j++)
        {
            cout<<c[i][j]<<' ';
        }
        cout<<endl;
    }
}
 int  main()
{
    long long n,m,l;
    while(cin>>n>>m)
    {
        in(n,m,a);
        cin>>m>>l;
        in(m,l,b);
        sout(n,m,l);
    }
}

4 結束語

    享和幫助是人生一大樂事,希望可以幫助您。本人才疏學淺,如果有不當之處,還請批評指正。同時歡迎大家評論、點贊及轉發!