从C ++中的函数返回向量或数组

时间:2016-05-06 05:10:14

标签: c++ arrays function matrix vector

我想创建一个程序,由于用户输入而创建N矩阵,但我需要返回向量(或数组)值,然后将它们发送到函数,并再次返回一个向量。 例如:

vector<int> createMatrix(rows, cols){
    for(int x = 0; x < rows; x++){
        for(int y = 0; y < cols; y++){
            cout << "Please, input the values for the row " << x << " and col " << y << endl;
            vector<int> Matrix(...)values;
            cin >> (...);
        }
    }
return Matrix; //here is the point
}

vector<int> mathOperations(operation, matrixA, matrixB){
    (...)
    return vector<int> resultMatrix;
}

int main(int argc, char *argv[]){
    int nMatrix, nRows, nCols;
    cout << "Please, answer how many matrix would you like do be created?" << endl;
    cin >> nMatrix;
    cout << "Thank you, and how many rows your matrix will have?" << endl;
    cin >> nRows;
    cout << "Thank you again, and how many cols?" << endl;
    cin >> nCols;
    cout << "Okey, creating " << nMatrix << " nMatrix for you!" << endl;

    for(int n = 0; n < nMatrix; n++){
        cout << "Please, insert the values for the matrix no" << n+1 << endl;
        vector<int> myMatrix[n] = createMatrix(nRows, nCols);
    }

    /* the rest of my code with functions to print values, create a new vectors by math operations between given arguments
    */

return 0;
}

最好的方法是什么?谢谢你提前。

1 个答案:

答案 0 :(得分:3)

如果您正在寻找使用vector构建二维结构的方法,请使用以下内容:

#include <vector>
#include <iostream>

using std::vector;
using std::cout;
using std::cin;

typedef vector<vector<int> > matrix_t;

matrix_t createMatrix(int rows, int cols){
  matrix_t Matrix(rows, vector<int>(cols));

  for(int x = 0; x < rows; x++){
    for(int y = 0; y < cols; y++){
      cout << "Please, input the values for the row "
        << x << " and col " << y << std::endl;
      cin >> Matrix[x][y];
    }
  }
  return Matrix;
}

int main(int argc, char const* argv[])
{
  matrix_t M(createMatrix(2, 2));

  for (int i = 0; i < 2; i++) {
    for (int j = 0; j < 2; j++) {
      cout << M[i][j] << std::endl;
    }

  }

  return 0;
}