[programmers] 배열의 길이를 2의 거듭제곱으로 만들기 - Java

문제 설명

제한사항

• 1 ≤ arr의 길이 ≤ 1,000

• 1 ≤ arr의 원소 ≤ 1,000

입출력 예

입출력 예 설명

• 예제 1번의 arr의 길이는 6입니다. arr의 길이를 2의 정수 거듭제곱으로 만드는 방법은 0을 2개, 10개, 26개,..., 추가하는 방법이 있고 그중 최소한으로 0을 추가하는 방법은 2개를 추가하는 것입니다. 따라서 [1, 2, 3, 4, 5, 6, 0, 0]을 return 합니다.

• 예제 2번의 arr의 길이는 4이고 이미 2의 정수 거듭제곱입니다. 따라서 뒤에 0을 추가하지 않아도 되므로 [58, 172, 746, 89]를 return 합니다.

solution.java

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        long powered = 2;
        int num = arr.length;
        
        for(int i: arr){
            list.add(i);
        }
        if(num == 1){
            return list.stream().mapToInt(i -> i).toArray();
        }else{
	        while (powered < num) {
	          powered = powered * 2;
	        }
	        for(int i=list.size(); i<powered; i++){
	            list.add(0);
	        }
	        return list.stream().mapToInt(i -> i).toArray();
				}
    }
}

핵심 키워드

• 조건 중 arr의 길이가 1일 경우도 있으므로 해당 경우엔 다른 수정을 하지 않고 배열을 리턴한다.

• 아니라면 배열의 길이보다 큰 2의 거듭제곱을 반복문을 통해 구하고, 해당 길이까지 0을 리스트의 뒤에 add한 후, 리스트를 배열로 바꿔 리턴한다.

결론!