프로그래머스 - 정사각형으로 만들기 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/181830

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr) {
    vector<vector<int>> answer;
    int row = arr.size();
    int col = arr[0].size();

    int gap = (row < col)? col - row : row - col;

    if(row < col)
    {
        for(int j=0;j<col-row;j++)
        {
            vector<int> a;
            for(int i=0;i<col;i++)
                a.push_back(0);
            arr.push_back(a);
        }
    }
    else
    {
        for(int i=0;i<row;i++)
        {
            for(int j=0;j<row-col;j++)
            {
                arr[i].push_back(0);
            }

        }
    }

    return arr;
}

 

설명

정사각형으로 만들기 위해 행과 열의 크기를 비교한다. 

만약 행이 더 긴 경우에는 각 arr[i] 마다 size가 행의 수가 될 때까지 0을 추가하면 된다.

그리고 열이 더 긴 경우에는 같은 크기의 열을 1개 만들어 삽입하는 방식으로 행의 수가 열의 길이가 될 때까지 반복하면 된다.