프로그래머스 - 배열의 원소만큼 추가하기 (Python ver.)

 

해결일	언어	레벨	테스트명
20240730	Python	level 0	배열의 원소만큼 추가하기

 

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

 

 

---

 

내 코드

def solution(arr):   
    return [i for i in arr for j in range(i)]

 

이중for문 연습했다.

 

이중for문 틀은 아래에 기재해둠!

https://young-1-2.tistory.com/133

Python 이중 for문 작성법

 

베스트 코드

def solution(arr):
    answer = []
    for num in arr:
        answer += [num] * num
    return answer

 

[] * 숫자 하면 기존 리스트에 추가되는 모양이다.

이것도 기억해두면 좋을 것 같다!