1. 程式人生 > 程式設計 >C++ 讓函式返回陣列的方法

C++ 讓函式返回陣列的方法

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

#include <iostream>

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,1.75 };
  float B[4] = {1,1,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 <iostream>

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

  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好了之後傳進去修改。

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

以上就是C++ 讓函式返回陣列的方法的詳細內容,更多關於C++ 讓函式返回陣列的資料請關注我們其它相關文章!