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

https://school.programmers.co.kr/learn/courses/30/lessons/181861

프로그래머스

 

 

 

 

정답 

 

import java.util.ArrayList;

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

 

 

다른 사람의 풀이 

 

1)

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;
    }
}