[프로그래머스] 배열의 원소만큼 추가하기

문제
아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 양의 정수 배열 arr가 매개변수로 주어질 때, arr의 앞에서부터 차례대로 원소를 보면서 원소가 a라면 X의 맨 뒤에 a를 a번 추가하는 일을 반복한 뒤의 배열 X를 return 하는 solution 함수를 작성해 주세요.

 

풀이
리스트로 배열을 만들어 그 배열의 길이내에서, 배열의 숫자만큼 저장한 후,
새로운 배열에 리스트의 길이만큼 저장후 출력

 import java.util.*;
class Solution {
    public int[] solution(int[] arr) {
        List<Integer> result = new ArrayList<>();
        
       for(int i = 0; i < arr.length; i++){
            for(int j = 0; j < arr[i]; j++){
                result.add(arr[i]);
            }
        }
        
        int[] answer = new int[result.size()];
        for(int i=0;i<result.size();i++){
            answer[i]=result.get(i);
        }
        return answer;
    }
}

 

다른사람 풀이

class Solution {
    public int[] solution(int[] arr) {
        int[] answer = {};

        int sum = 0;

        for(int i=0; i<arr.length; i++) {
            sum += arr[i];    
        }

        answer = new int[sum];

        int idx = 0;

        for(int i=0; i<arr.length; i++) {
            for(int j=0; j<arr[i]; j++) {
                answer[idx++] = arr[i];
            }
        }

        return answer;
    }
}