[Programmers] Lv.0 / 배열의 원소만큼 추가하기 / Java

문제

 

풀이

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr) {
        
        int length = Arrays.stream(arr).sum();
        int[] result = new int[length];
        int index = 0;
        
        for(int num : arr) {
            for(int i = 0; i < num; i++) {
                result[index++] = num;
            }
        }
        return result;
    }
}

 

출처

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