[프로그래머스/파이썬] Lv.0 빈 배열에 추가, 삭제하기 (18')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-20

소요 시간 : 18min

 

---

 

문제 설명

아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 길이가 같은 정수 배열 arr과 boolean 배열 flag가 매개변수로 주어질 때, flag를 차례대로 순회하며 flag[i]가 true라면 X의 뒤에 arr[i]를 arr[i] × 2 번 추가하고, flag[i]가 false라면 X에서 마지막 arr[i]개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.

 

 

 

나의 풀이

def solution(arr, flag):
    X = []
    for a, f in zip(arr, flag) :
        if f == True :
            X.extend([a] * (a*2))
        else :
            del X[-a:]
    return X