C++ 怎樣讓函數返回數組


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

實例

 1 #include <iostream>
 2 
 3 using namespace std;  4 
 5 float* MultMatrix(float A[4], float B[4])  6 {  7     float M[4];  8     M[0] = A[0]*B[0] + A[1]*B[2];  9     M[1] = A[0]*B[1] + A[1]*B[3]; 10     M[2] = A[2]*B[0] + A[3]*B[2]; 11     M[3] = A[2]*B[1] + A[3]*B[3]; 12 
13     return M; 14 } 15 
16 int main() 17 { 18     float A[4] = { 1.75, 0.66, 0, 1.75 }; 19     float B[4] = {1, 1, 0, 0}; 20     float *M = MultMatrix(A, B); 21     cout << M[0] << " " << M[1] << endl; 22     cout << M[2] << " " << M[3] << endl; 23 
24     return 0; 25 }

但是運行后發現結果是:

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數組是一樣的,因為它們指向的是同一片內存空間。於是代碼就修改為:

實例

 1 #include <iostream>
 2 
 3 using namespace std;  4 
 5 float* MultMatrix(float A[4], float B[4])  6 {  7     float *M = new float[4];  8     M[0] = A[0]*B[0] + A[1]*B[2];  9     M[1] = A[0]*B[1] + A[1]*B[3]; 10     M[2] = A[2]*B[0] + A[3]*B[2]; 11     M[3] = A[2]*B[1] + A[3]*B[3]; 12     cout << M[0] << " " << M[1] << endl; 13     cout << M[2] << " " << M[3] << endl; 14 
15     return M; 16 } 17 
18 int main() 19 { 20     float A[4] = { 1.75, 0.66, 0, 1.75 }; 21     float B[4] = {1, 1, 0, 0}; 22     float *M = MultMatrix(A, B); 23     cout << M[0] << " " << M[1] << endl; 24     cout << M[2] << " " << M[3] << endl; 25     delete[] M; 26 
27     return 0; 28 }

運行結果:

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; }

點評內容:

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

其次,C++ 里面手動內存分配的一個重要原則是誰分配誰釋放。

所以,不應該在MultMatrix里new數組,而應該在外面new好了之后傳進去修改。

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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM