[연습문제] 푸드 파이트 대회

https://school.programmers.co.kr/learn/courses/30/lessons/134240

프로그래머스

---

문제 설명
수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다.

이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다.

대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다.

중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.

이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다.

또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다.

이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.

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

따라서 1번 음식 1개는 대회에 사용하지 못합니다.

수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.

---

제한사항

• 2 ≤ food의 길이 ≤ 9
• 1 ≤ food의 각 원소 ≤ 1,000
• food에는 칼로리가 적은 순서대로 음식의 양이 담겨 있습니다.
• food[i]는 i번 음식의 수입니다.
• food[0]은 수웅이가 준비한 물의 양이며, 항상 1입니다.
• 정답의 길이가 3 이상인 경우만 입력으로 주어집니다.

---

입출력 예

food	result
[1, 3, 4, 6]	"1223330333221"
[1, 7, 1, 2]	"111303111"

---

입출력 예 설명
입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 두 선수는 1번 음식 3개, 3번 음식 1개를 먹게 되므로 음식의 배치는 "111303111"입니다.

---

푸드 파이팅 대회를 위해, 2명의 선수가 먹을 음식들을 서로 공평하게 배치하는 방법을 구하는 문제입니다.

 

이 문제의 특징이라면 항상 2명의 선수가 참가하기 때문에, 물(0)을 기준으로 보았을 때 좌우 대칭 구조라는 점입니다.

그리고 배치될 수 있는 음식의 개수는 총 개수의 절반, 혹은 절반 - 1이라는 점이죠.

따라서 물을 제외한 모든 음식은 항상 짝수 개만큼만 사용될 수 있습니다.

 

대칭 구조이기 때문에 왼쪽에 위치한 선수가 먹을 수 있도록 배치를 구성한 뒤, 0을 추가하고 대칭되도록 만들어주면 됩니다.

이 문제는 스택 구조를 사용할 수도 있을 것 같네요.

자바로 풀 때는 스택을, 파이썬으로 풀 때는 리스트 슬라이싱을 사용해 풀어보겠습니다.

 

정답 코드를 작성해보겠습니다.

 

1. 자바

import java.util.Stack;

class Solution {
    public String solution(int[] food) {
        // 배치를 저장할 answer
        String answer = "";
        // 스택의 사이즈를 저장할 size
        int size = 0;
        // 스택을 선언합니다.
        Stack<Integer> st = new Stack<Integer>();
        // 준비된 음식들에 대해서 탐색합니다.
        for (int i = 1; i < food.length; i++) {
            // 사용량 = 준비된 음식의 절반
            int usable = food[i] / 2;
            // 해당 사용량만큼 스택에 추가합니다.
            for (int j = 0; j < usable; j++) {
                st.push(i);
                answer += i;
            }
        }
        // 스택 크기 설정
        size = st.size();
        // 중간 물 추가
        answer += 0;
        // 스택에서 빼내면서 값을 추가합니다.
        for (int i = 0; i < size; i++) {
            answer += st.pop();
        }

        return answer;
    }
}

 

2. 파이썬

def solution(food):
    # 배치를 저장할 answer
    answer = ''
    # 물을 제외하고 리스트 값을 가져옵니다.
    for i, item in enumerate(food[1:]):
        # 가능한 사용량 = 절반
        usage = item // 2
        # 배치에 사용량만큼 해당 인덱스를 추가합니다.
        answer += str(i + 1) * usage
    # 마지막에 0을 추가하고, 뒤집은 배치를 추가합니다.
    answer = answer + '0' + answer[::-1]
    return answer

 

최종적인 배치가 좌우 대칭인 구조라는 점만 기억하면 편하게 풀 수 있습니다.