• 문제 설명

for idx, i in zip(enumerate(arr, flag)):

코드에서 문제

for idx, i in zip(enumerate(arr, flag)):
    ...

여기서 enumerate(arr, flag) 자체가 잘못됐어.

• enumerate(arr, start=flag)처럼 쓴 거라서 flag 리스트를 통째로 "시작 인덱스"로 잘못 넘기고 있는 거야.
• 그래서 TypeError 날 확률이 높아.

<aside> 💡

zip(arr, flag) 이렇게 사용해야 함!!

</aside>

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

• 📌 Python zip & 슬라이싱 주의사항