藍橋杯 矩陣乘法 C++算法提高 HERODING的藍橋杯之路

資源限制
時間限制:1.0s 內存限制:256.0MB
問題描述
  小明最近剛剛學習了矩陣乘法,但是他計算的速度太慢,於是他希望你能幫他寫一個矩陣乘法的運算器。
輸入格式
  輸入的第一行包含三個正整數N,M,K,表示一個NM的矩陣乘以一個MK的矩陣。接下來N行,每行M個整數,表示第一個矩陣。再接下來的M行,每行K個整數,表示第二個矩陣。
輸出格式
  輸出有N行,每行K個整數,表示矩陣乘法的結果。
樣例輸入
2 1 3
1
2
1 2 3
樣例輸出
1 2 3
2 4 6
數據規模和約定
  0<N,M,K<=100, 0<=矩陣中的每個數<=1000。

解題思路:
這是一道中規中矩的矩陣乘法題,考驗的是二重for循環和三重for循環的運用,還有線性代數的知識,一定要注意乘法的細節,以及循環時的限定範圍,代碼如下:

#include<bits/stdc++.h>

using namespace std;

int a[100][100];
int b[100][100];
int c[100][100];

int main(){
	int N, M, K;
	cin >> N >> M >> K;
	int i, j, k;
	for(i = 0; i < N; i ++){
		for(j = 0; j < M; j ++){
			cin >> a[i][j];
		}
	}
	for(i = 0; i < M; i ++){
		for(j = 0; j < K; j ++){
			cin >> b[i][j];
		}
	}
	for(i = 0; i < N; i ++){
		for(j = 0; j < K; j ++){
			int temp = 0;
			for(k = 0 ; k < M; k ++){
				temp += a[i][k] * b[k][j];
			}
			c[i][j] = temp;
		}
	}
	for(i = 0; i < N; i ++){
		for(j = 0; j < K; j ++){
			cout << c[i][j] << " ";
		}
		cout << endl;
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章