千張筆記

題目：實(shí)現(xiàn)兩個(gè)N*N矩陣的乘法，矩陣由一維數(shù)組表示。

答案：

#include <iostream>
using namespace std;

#define  SIZE 2

int* multi(int* array_a,int* array_b,int N)
{
    int* array_c = (int*)malloc(N*N);
    int temp;
    for (int i = 0; i < N; ++ i)
        for(int j = 0; j < N; ++ j)
    {
        temp = i * N + j;
        *(array_c + temp) = 0;
        for (int k = 0; k < N; ++ k)
        {
            *(array_c + temp) += array_a[i * N + k] * array_b[k * N + j];
        }
        cout << *(array_c + temp) << " ";
        
    }
    return array_c;
}

int main()
{
    int array_a[SIZE * SIZE] = {1,2,1,2};
    int array_b[SIZE * SIZE] = {2,1,2,1};
    int* array_c = multi(array_a,array_b,SIZE);
    free(array_c);
    return 0;

}

注意：malloc或new分配內(nèi)存后，要用free或delete釋放內(nèi)存，要不然可能有“_CrtIsValidHeapPointer異常”，也會(huì)有內(nèi)存泄露。

From：http://hi.baidu.com/feigudian/blog/item/ac608c013c81d0d5277fb5ee.html