怎样让函数返回数组

这个问题属于非常初级的问题,但是对于初学不知道的人可能会比较头疼。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一个数组,这样就不会被释放掉了。

于是就应该将

7floatM[4];

改为:

7 float *M = new float[4];

修改运行后得到结果:

1.75 1.75

0 0

1.75 1.75

0 0

正确。但是我们这样并没有将自己申请的空间释放掉,如果我们在函数内释放的话结果就会跟开始时的一样了。

看看我们的调用代码:

20float*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 }

至于数组类和智能智能我还需要继续学习,多谢大家关注!
原文链接: https://www.cnblogs.com/yangxi/archive/2011/09/18/2180759.html

欢迎关注

微信关注下方公众号,第一时间获取干货硬货;公众号内回复【pdf】免费获取数百本计算机经典书籍

原创文章受到原创版权保护。转载请注明出处:https://www.ccppcoding.com/archives/32502

非原创文章文中已经注明原地址,如有侵权,联系删除

关注公众号【高性能架构探索】,第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处!

(0)
上一篇 2023年2月8日 上午9:44
下一篇 2023年2月8日 上午9:45

相关推荐