Level 0. 정사각형으로 만들기

Level 0. 정사각형으로 만들기

 

 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr)
{
    int row = arr.size(), col = arr.front().size();

    if (row < col)
    {
        arr.resize(col);

        fill(arr.begin() + row, arr.end(), vector<int>(col, 0));
    }
    else if (row > col)
    {
        for_each(arr.begin(), arr.end(), [&](vector<int>& v)
        {
            v.resize(row);

            fill(v.begin() + col, v.end(), 0);
        });
    }

    return arr;
}