怎样让函数返回数组
这个问题属于非常初级的问题,但是对于初学不知道的人可能会比较头疼。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 }
至于数组类和智能智能我还需要继续学习,多谢大家关注!
作者“ 杨溪”
补充:软件开发 , C语言 ,