😈 알고리즘/🖥️ 프로그래머스

🖥️ 2차원으로 만들기

Buᐢ༝ᐢy 2023. 1. 25. 16:42

코딩테스트 연습 - 2차원으로 만들기

문제 설명


정수 배열 num_list와 정수 n이 매개변수로 주어집니다. num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.

num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 num_list를 2 * 4 배열로 다음과 같이 변경합니다. 2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.

num_list n result

[1, 2, 3, 4, 5, 6, 7, 8] 2 [[1, 2], [3, 4], [5, 6], [7, 8]]

제한사항


  • num_list의 길이는 n의 배 수개입니다.
  • 0 ≤ num_list의 길이 ≤ 150
  • 2 ≤ n < num_list의 길이

입출력 예


num_list n result

[1, 2, 3, 4, 5, 6, 7, 8] 2 [[1, 2], [3, 4], [5, 6], [7, 8]]
[100, 95, 2, 4, 5, 6, 18, 33, 948] 3 [[100, 95, 2], [4, 5, 6], [18, 33, 948]]

입출력 예 설명


입출력 예 #1

  • num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 2 * 4 배열로 변경한 [[1, 2], [3, 4], [5, 6], [7, 8]] 을 return합니다.

입출력 예 #2

  • num_list가 [100, 95, 2, 4, 5, 6, 18, 33, 948] 로 길이가 9이고 n이 3이므로 3 * 3 배열로 변경한 [[100, 95, 2], [4, 5, 6], [18, 33, 948]] 을 return합니다.

제출 코드


2022년 12월 18일

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<int> num_list, int n) {

    int length = num_list.size() / n;

    vector<vector<int>> answer(length - 1);
    answer.push_back(vector<int>());

    for (int i = 0; i < length; i++)
    {
        for (int j = 0; j < n; j++)
        {
            answer[i].push_back(num_list[i * n + j]);
        }
    }

    return answer;
}

<aside> 1️⃣ 제한 사항을 보면 문제 접근하기가 쉽다.

  • num_list의 길이는 n의 배 수개입니다. → length를 전체 크기에서 n으로 나눈 이유이다.

</aside>

<aside> 2️⃣ 2번 예시로 보면 length는 3이니 i로 배열 순서를 넣어준다.

  • answer[i].push_back(num_list[i * n + j]);

</aside>

<aside> 3️⃣ 0번째에서 0번, 1번, 2번을 0 * 3 + 0, 0 * 3 +1, 0 * 3 +2가 되니 값이 순서대로 알맞게 잘 들어간다.

</aside>