[프로그래머스/Java] Lv.0 배열의 원소만큼 추가하기

문제

 

 

728x90

---

풀이

 

 

 

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