This commit is contained in:
2020-03-29 21:05:11 -06:00
parent 6a940a7519
commit 7f98eb207f
2 changed files with 5 additions and 5 deletions

View File

@@ -19,11 +19,11 @@ namespace matrixTools
} }
template <typename T> template <typename T>
std::vector<T> FlattenMatrix(std::vector<std::vector<T>> matrix) std::vector<T> FlattenMatrix(std::vector<std::vector<T>> matrix, int columnSize, int rowSize)
{ {
std::vector<T> rVector(matrix.size() * matrix[0].size(), 0); std::vector<T> rVector((rowSize*columnSize), 0);
for (int i = 0; i < matrix.size(); i++) { for (int i = 0; i < columnSize; i++) {
for (int j = 0; j < matrix[0].size(); j++) { for (int j = 0; j < rowSize; j++) {
rVector[i + j] = matrix[i][j]; rVector[i + j] = matrix[i][j];
} }
} }

View File

@@ -28,7 +28,7 @@ int main(int argc, char* argv[])
} }
std::cout << std::endl; std::cout << std::endl;
} }
flatMatrix = matrixTools::FlattenMatrix(matrix); flatMatrix = matrixTools::FlattenMatrix(matrix, matrix.size(), matrix[0].size());
for(int i = 0; i < flatMatrix.size(); i++){ for(int i = 0; i < flatMatrix.size(); i++){
std::cout << flatMatrix[i] << " "; std::cout << flatMatrix[i] << " ";
} }