코딩 테스트 (Java)/코딩 기초 트레이닝 (프로그래머스)

181861. 배열의 원소만큼 추가하기

가지코딩 2025. 4. 22. 18:03

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr


문제

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

 

제한사항

  • 1 ≤ arr의 길이 ≤ 100
  • 1 ≤ arr의 원소 ≤ 100

 

입출력 예

arr result
[5, 1, 4] [5, 5, 5, 5, 5, 1, 4, 4, 4, 4]
[6, 6] [6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]
[1] [1]

 

입출력 예 #1

  • 예제 1번에 대해서 a와 X를 나타내보면 다음 표와 같습니다.
a X
  []
5 [5, 5, 5, 5, 5]
1 [5, 5, 5, 5, 5, 1]
4 [5, 5, 5, 5, 5, 1, 4, 4, 4, 4]
따라서 [5, 5, 5, 5, 5, 1, 4, 4, 4, 4]를 return 합니다.

 

입출력 예 #2

  • 예제 2번에 대해서 a와 X를 나타내보면 다음 표와 같습니다.
a X
  []
6 [6, 6, 6, 6, 6, 6]
6 [6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]
따라서 [6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]를 return 합니다.

 

입출력 예 #3

  • 예제 2번에 대해서 a와 X를 나타내보면 다음 표와 같습니다.
a X
  []
1 [1]
따라서 [1]을 return 합니다.

풀이

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        
        for (int a : arr) {
            for (int i = 0; i < a; i++) {
                list.add(a);
            }
        }
        
        return list.stream().mapToInt(i -> i).toArray();
    }
}

 

 

 

 


GitHub: https://github.com/gajicoding/coding_test/edit/main/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4/0/181861.%E2%80%85%EB%B0%B0%EC%97%B4%EC%9D%98%E2%80%85%EC%9B%90%EC%86%8C%EB%A7%8C%ED%81%BC%E2%80%85%EC%B6%94%EA%B0%80%ED%95%98%EA%B8%B0


출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

 

코딩테스트 연습 | 프로그래머스 스쿨

개발자 취업의 필수 관문 코딩테스트를 철저하게 연습하고 대비할 수 있는 문제를 총망라! 프로그래머스에서 선발한 문제로 유형을 파악하고 실력을 업그레이드해 보세요!

school.programmers.co.kr