Passing two-dimensional array to a function using pointer #2

標題：使用指標傳值的方式做矩陣相乘運算
VC++ (main.cpp)：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51

#include "stdafx.h" #include <iostream> using namespace std; void multiply(float *matrixA, float *matrixB, float *matrixC, int p, int q, int r) { int i = 0, j = 0, k = 0; float sum = 0.0f; for(i=0;i<p;i++) { for(j=0;j<r;j++) { sum = 0.0f; for (k = 0; k < q; k++) { sum += *(matrixA + i*q + k) * *(matrixB + k*r + j); } *(matrixC + i*r + j) = sum; } } } int _tmain(int argc, _TCHAR* argv[]) { const int p = 4, q = 2, r = 3; int i = 0, j = 0; float matrixA[p][q] = { //4x2矩陣 {3, -1}, {0, 3}, {1, 4}, {2, 1} }; float matrixB[q][r] = { //2x3矩陣 {1, -2, 3}, {2, 1, 0} }; float matrixC[p][r] = {}; //4x3矩陣 multiply(&matrixA[0][0], &matrixB[0][0], &matrixC[0][0], p, q, r); //印出二維陣列資料 for(i=0;i<p;i++) { for(j=0;j<r;j++) { cout << matrixC[i][j] << "\t"; } cout << endl; } return 0; }

說明：
將3個二維陣列初始位址及陣列大小傳入函式，並在函式中針對指標內容進行矩陣相乘運算操作。