函數返回數組，返回數組

函數返回數組，返回數組

這個問題屬於非常初級的問題，但是對於初學不知道的人可能會比較頭疼。C++中函數是不能直接返回一個數組的，但是數組其實就是指針，所以可以讓函數返回指針來實現。比如一個矩陣相乘的函數，很容易地我們就寫成：

#include <iostream>

using namespace std;

float* MultMatrix(float A[4], float B[4])
{
    float M[4];
    M[0] = A[0]*B[0] + A[1]*B[2];
    M[1] = A[0]*B[1] + A[1]*B[3];
    M[2] = A[2]*B[0] + A[3]*B[2];
    M[3] = A[2]*B[1] + A[3]*B[3];

    return M;
}

int main()
{
    float A[4] = { 1.75, 0.66, 0, 1.75 };
    float B[4] = {1, 1, 0, 0};
    float *M = MultMatrix(A, B);
    cout << M[0] << " " << M[1] << endl;
    cout << M[2] << " " << M[3] << endl;

    return 0;
}

但是運行後發現結果是：1.75 1.75
                               6.51468e-039 3.76489e-039

根本不是想要的結果。於是我們在函數中也加上顯示代碼，看看是不是計算的問題，得到結果：

1.75 1.75
0 0
1.75 1.75
1.96875 1.75

發現計算的結果是正確的，但返回後就變了，而且跟上次的結果不一樣。這是為什麼呢？

因為在函數中定義的數組M在函數執行完後已經被系統釋放掉了，所以在調用函數中得到的結果當然不是計算後的結果。有一個解決辦法就是動態分配內存，在函數中new一個數組，這樣就不會被釋放掉了。

於是就應該將

7 　　 float M[4];

改為：

7　　float *M = new float[4];

修改運行後得到結果：

1.75 1.75
0 0
1.75 1.75
0 0

正確。但是我們這樣並沒有將自己申請的空間釋放掉，如果我們在函數內釋放的話結果就會跟開始時的一樣了。

看看我們的調用代碼：

20 　　 float *M = MultMatrix(A, B);

這樣其實是將M指針指向了函數中M數組的首地址，我們可以將M指針釋放，效果和釋放申請的M數組是一樣的，因為它們指向的是同一片內存空間。於是代碼就修改為：

#include <iostream>

using namespace std;

float* MultMatrix(float A[4], float B[4])
{
    float *M = new float[4];
    M[0] = A[0]*B[0] + A[1]*B[2];
    M[1] = A[0]*B[1] + A[1]*B[3];
    M[2] = A[2]*B[0] + A[3]*B[2];
    M[3] = A[2]*B[1] + A[3]*B[3];
    cout << M[0] << " " << M[1] << endl;
    cout << M[2] << " " << M[3] << endl;

    return M;
}

int main()
{
    float A[4] = { 1.75, 0.66, 0, 1.75 };
    float B[4] = {1, 1, 0, 0};
    float *M = MultMatrix(A, B);
    cout << M[0] << " " << M[1] << endl;
    cout << M[2] << " " << M[3] << endl;
    delete[] M;

    return 0;
}

運行結果：

1.75 1.75
0 0
1.75 1.75
0 0
沒有問題，new的空間也delete掉了。

當然，最好是在外面申請內存，然後再傳給函數：

#include <iostream>

using namespace std;

void MultMatrix(float M[4], float A[4], float B[4])
{
    M[0] = A[0]*B[0] + A[1]*B[2];
    M[1] = A[0]*B[1] + A[1]*B[3];
    M[2] = A[2]*B[0] + A[3]*B[2];
    M[3] = A[2]*B[1] + A[3]*B[3];

    cout << M[0] << " " << M[1] << endl;
    cout << M[2] << " " << M[3] << endl;
}

int main()
{
    float A[4] = { 1.75, 0.66, 0, 1.75 };
    float B[4] = {1, 1, 0, 0};

    float *M = new float[4];
    MultMatrix(M, A, B);

    cout << M[0] << " " << M[1] << endl;
    cout << M[2] << " " << M[3] << endl;
    delete[] M;

    return 0;
}