天天看點

html函數傳回一個數組,C++ 怎樣讓函數傳回數組

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

執行個體

#include

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 一個數組,這樣就不會被釋放掉了。

于是就應該将:

float M[4];

改為:

float *M = new float[4];

修改運作後得到結果:

1.75 1.75

0 0

1.75 1.75

0 0

正确。但是我們這樣并沒有将自己申請的空間釋放掉,如果我們在函數内釋放的話結果就會跟開始時的一樣了。

看看我們的調用代碼:

float *M = MultMatrix(A, B);

這樣其實是将M指針指向了函數中M數組的首位址,我們可以将M指針釋放,效果和釋放申請的M數組是一樣的,因為它們指向的是同一片記憶體空間。于是代碼就修改為:

執行個體

#include

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

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;

}

點評内容:

首先,數組的 delete 是 delete[]。

其次,C++ 裡面手動記憶體配置設定的一個重要原則是誰配置設定誰釋放。

是以,不應該在MultMatrix裡new數組,而應該在外面new好了之後傳進去修改。

要想傳回一個數組,使用智能指針之類的東西才是正途。