[파이썬 코딩 테스트] 푸드 파이트 대회

ch.2·2024년 8월 12일

코딩 테스트

목록 보기

17/21

문제

프로그래머스 연습 문제 Lv.1

수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다. 대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다. 중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.
이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다. 또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다. 이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.
수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.

문제 예시
예를 들어, 3가지의 음식이 준비되어 있으며, 칼로리가 적은 순서대로 1번 음식을 3개, 2번 음식을 4개, 3번 음식을 6개 준비했으며, 물을 편의상 0번 음식이라고 칭한다면, 두 선수는 1번 음식 1개, 2번 음식 2개, 3번 음식 3개씩을 먹게 되므로 음식의 배치는 "1223330333221"이 됩니다. 따라서 1번 음식 1개는 대회에 사용하지 못합니다.

첫 번째 답안

def solution(food):
    answer = []
    for i in range(len(food)):
        if i  == 0:
            answer.append(food[i])
        elif i % 2 == 0:
            answer.append(str(i)* (food[i]//2))
        else:
            answer.append('임시')
    return answer

food[i]의 값을 2로 나눠서 양쪽 끝에 배치하고자 했다.
이를 위해 i 의 값을 문자열로 변환해 2로 나눠지는 몫만큼 반복해 answer 리스트에 넣었다.
cf) str(a) * 3 = aaa

하지만 food[i]가 홀수인 경우에는 연산할 수 없었고, 짝수인 food[i]를 리스트 혹은 문자열의 양쪽 끝으로 정렬하는 방법은 찾지 못했다.

→ food[i]의 순서대로 추가하거나 정렬하지 말고, 대칭 중 앞 부분만 먼저 완성한 후, 뒷 부분을 붙이자.

두 번째 답안

🥚🥚🥚

def solution(food):
    answer = []
    for i in range(len(food)):
        if i > 0:
            answer.append(str(i) * (food[i] // 2))
    return answer

대칭으로 이어붙일 것이니까
if i == 0은 삭제하고, if i > 0 구문을 넣었다.
food[i]가 홀수인 경우에 2로 나누었을 때 생기는 나머지는 대회에 사용하지 못하고 남는 음식이 되므로 신경쓰지 않는다.

0을 기준으로 앞 부분에 배치할 요소가 정해졌다.

🐣🐣🐣

def solution(food):
    answer = []
    for i in range(len(food)):
        if i > 0:
            answer.append(str(i) * (food[i] // 2))
    answer = ''.join(answer)
    return answer

answer 리스트 안 요소를 문자열로 이어붙이는 .join( ) 메서드를 사용해 재정의했다.

대칭 앞 부분이 완성되었다.

🐥🐥🐥

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