VC++ (main.cpp):
#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個二維陣列初始位址及陣列大小傳入函式,並在函式中針對指標內容進行矩陣相乘運算操作。
沒有留言:
張貼留言