這個問題屬於非常初級的問題,但是對於初學不知道的人可能會比較頭疼。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一個數組,這樣就不會被釋放掉了。
於是就應該將
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數組是一樣的,因為它們指向的是同一片內存空間。於是代碼就修改為:
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掉了。
鑒於下面大牛們的建議,我將程序修改如下,大家看可否:
1 #include <iostream>
2
3 using namespace std;
4
5 void MultMatrix(float M[4], float A[4], float B[4])
6 {
7 M[0] = A[0]*B[0] + A[1]*B[2];
8 M[1] = A[0]*B[1] + A[1]*B[3];
9 M[2] = A[2]*B[0] + A[3]*B[2];
10 M[3] = A[2]*B[1] + A[3]*B[3];
11
12 cout << M[0] << " " << M[1] << endl;
13 cout << M[2] << " " << M[3] << endl;
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
21 float *M = new float[4];
22 MultMatrix(M, A, B);
23
24 cout << M[0] << " " << M[1] << endl;
25 cout << M[2] << " " << M[3] << endl;
26 delete[] M;
27
28 return 0;
29 }
至於數組類和智能智能我還需要繼續學習,多謝大家關注!
作者“ 楊溪”