[Programmers][Python] Lv.1 푸드 파이트 대회

[문제]

https://school.programmers.co.kr/learn/courses/30/lessons/134240?language=python3

프로그래머스

 

 

[코드]

def solution(food):
    pre_food, arr = [], []
    for i in range(1,len(food)):
        if food[i]//2 >= 1:
            arr.append(i)
            pre_food.append(food[i]//2)

    answer = ''
    for i in range(len(arr)):
        answer += str(arr[i]) * pre_food[i]

    answer = answer + '0' + answer[::-1]
    return answer

 

정리한 코드

def solution(food):
    answer = ''
    for i in range(1,len(food)):
        if food[i]//2 >= 1:
            answer += str(i) * (food[i]//2)
    
    return answer + '0' + answer[::-1]