[C++] 프로그래머스 배열의 길이를 2의 거듭제곱으로 만들기

문제설명

정수 배열 arr이 매개변수로 주어집니다.
arr의 길이가 2의 정수 거듭제곱이 되도록
arr 뒤에 정수 0을 추가하려고 합니다.
arr에 최소한의 개수로 0을 추가한 배열을
return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000

입출력 예

arr	result
[1, 2, 3, 4, 5, 6]	[1, 2, 3, 4, 5, 6, 0, 0]
[58, 172, 746, 89]	[58, 172, 746, 89]

풀이

정수 배열 arr의 길이가 2의 정수 거듭제곱이 되도록 로직을 구현하는 구현 문제입니다.

#include <string>
#include <vector>
#include <cmath>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer = arr;
    int i = 0;
    
    while(pow(2, i) < arr.size())
    {
        i++;
    }
    
    int total_len = pow(2, i);
    
    for(int j = arr.size(); j < total_len; j++)
    {
        answer.push_back(0);
    }
    
    return answer;
}

결과