[프로그래머스] 정사각형으로 만들기

 

• 풀이

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr) {
    vector<vector<int>> answer;
    int max_s;
    if(arr[0].size() < arr.size()) max_s = arr.size();
    else max_s = arr[0].size();
    for(vector<int>& v : arr){
        v.resize(max_s,0);
    }
    while(arr.size() < max_s){
        arr.push_back(vector<int>(max_s,0));
    }
    return arr;
}