본문 바로가기
Coding Test/Java Coding Test

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

by nam_ji 2023. 12. 5.

배열의 원소만큼 추가하기

문제

  • 양의 정수 배열 arr이 매개변수로 주어집니다.
  • arr의 각 자리의 원소의 크기만큼 그 원소를 answer에 담아서 재 출력하는 문제입니다.


테스트 (이클립스)

  • 2중 for문을 활용해 봤습니다.
  • 처음 for문에는 arr의 크기만큼 순회하도록 하고
  • 두 번째 for문은 arr의 각 원소의 크기만큼 순회하도록 하였습니다.
  • 그 안에서 각 원소가 그 크기만큼 새로운 배열에 담아져야 하기 때문에 answer에 arr을 첫 번째 for문의 값을 가져오도록 하였습니다.
package com.namji.codingtest;

import java.util.ArrayList;
import java.util.List;

public class AddElements_as_ManyArrayElements {
	public static void main(String[] args) {
		/*
		 * 양의 정수 arr이 매개변수로 주어집니다.
		 * arr의 각 자리의 원소의 수만큼
		 * 각 자리의 원소를 늘려서 출력하는 문제입니다.
		 * ex) [5, 1, 4] > [5, 5, 5, 5, 5, 1, 4, 4, 4, 4]
		 * ex) [6, 6] > [6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]
		 * ex) [1] > [1]
		 */
		int[] arr = {5, 1, 4};
		List<Integer> answer = new ArrayList<Integer>();
		for (int i = 0; i < arr.length; i++) {
			for (int j = 0; j < arr[i]; j++) {
				answer.add(arr[i]);
			}
		}
		System.out.println(answer);
	}
}

프로그래머스 코드