C++에서 행렬 재구성

다른 플랫폼에는 'reshape'라는 매우 유용한 기능이 있습니다. 이 기능은 행렬을 크기가 다른 새로운 행렬로 재구성하는 데 사용되지만 데이터는 동일합니다. 따라서 원하는 재구성 행렬의 행 번호와 열 번호에 대해 행렬과 두 개의 값 r과 c가 있는 경우

따라서 입력이 [[5,10],[15,20]], row =1 및 col =4와 같으면 출력은 [[5, 10, 15, 20]]

이 문제를 해결하기 위해 다음 단계를 따르겠습니다-

• 어레이 온도 정의

• 크기(r x c)의 2D 배열 하나를 정의합니다.

• 개수 :=0

• initialize i :=0의 경우, i

  • j 초기화의 경우:=0, j

    • temp의 끝에 nums[i, j] 삽입

• r * c가 숫자의 크기와 같지 않으면 -

  • 숫자 반환

• initialize i :=0의 경우, i

  • initialize j :=0의 경우 j

    • 개수 =개수 + 1

    • res[i, j] :=temp[count]

• 반환 해상도

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<auto>> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << "[";
      for(int j = 0; j <v[i].size(); j++){
         cout << v[i][j] << ", ";
      }
      cout << "],";
   }
   cout << "]"<<endl;
}
class Solution {
public:
   vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
      vector<int> temp;
      vector<vector<int> > res(r, vector<int>(c));
      int count = 0;
      for (int i = 0; i < nums.size(); i++) {
         for (int j = 0; j < nums[0].size(); j++) {
            temp.push_back(nums[i][j]);
         }
      }
      if (r * c != nums.size() * nums[0].size())
         return nums;
      for (int i = 0; i < r; i++) {
         for (int j = 0; j < c; j++) {
            res[i][j] = temp[count++];
         }
      }
      return res;
   }
};
main(){
   Solution ob;
   vector<vector<int>> v = {{5,10},{15,20}};
   print_vector(ob.matrixReshape(v, 1, 4));
}

입력

{{5,10},{15,20}}, 1, 4

출력

[[5, 10, 15, 20, ],]