Show toolbar

2013年7月10日 星期三

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

標題:使用指標傳值的方式做矩陣相乘運算
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個二維陣列初始位址及陣列大小傳入函式,並在函式中針對指標內容進行矩陣相乘運算操作。

沒有留言:

張貼留言