[programmers] [Kotlin] 푸드 파이트 대회

문제 설명

수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 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"

코드

class Solution {
    fun solution(food: IntArray) = 
        food.mapIndexed {idx, it -> idx to it / 2}.subList(1, food.size).run{
            var answer = ""
            forEach{ pair ->
                for(i in 0 until pair.second) {
                    answer += pair.first.toString()
                }
            }
            answer + "0" + answer.reversed()
        }
}

풀이

우선 제공되는 배열인 food는 '음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열'이라 하였으므로 정렬은 신경 쓰지 않는다. 그리고 제한조건에 'food[0]은 수웅이가 준비한 물의 양이며, 항상 1입니다.' 이라했으므로 배열의 첫번째 요소는 신경쓰지 않는다.

 

제일 먼저 해야할 것은 food배열의 요소를 모두 2로 나누는 것이다. 두 선수가 먹는 음식의 양이 같아야 한다는 조건때문이다. 그리고 최종 출력은 음식의 순번을 나타내야하므로 mapIndexed를 통해 음식 순번과 음식의 양은 가진 Pair 형태로 구조를 바꿔준다. 바꿔주게 되면 1번 예시를 기준이로 food배열이 [(0,1), (1,1), (2,2), (3,3)] 형식으로 바뀐다.

 

이제 여기서 0번째 요소는 물이므로 음식 계산에 있어 필요 없으므로 subList를 통해 0번째 요소를 제외시켜준다. 

그 후 최종 문자열 출력을 위해, Scope함수를 통해서 필터링된 리스트를 조작해준다.

 

먼저, 최종 문자열을 담을 변수를 선언한다.

그리고 subList된 [(1,1), (2,2), (3,3)] 배열을 for문을 돌려 문자열을 만드는데 문자열을 만드는 로직은 다음과 같다.

음식의 갯수 (각 요소마다 두번째 요소) 대로 음식의 순번을 출력해야하므로 pair.second만큼 반복시키고 answer에는 pair.first를 반복횟수만큼 추가시켜준다.

 

위 과정을 마치면 우선 한 명의 음식 배치 순서를 만들 수 있게 된다. 양 선수 모두 같은 순서를 배정받아야하므로 가운데에 물인 0을 추가한 후, 반대 선수도 같은 순서로 대칭될 수 있게 reversed로 문자를 뒤집어 주어 합치면 최종적인 음식배치 문자열을 구할 수 있다.

결과

안정적으로 통과한 것을 볼 수 있다.