알고리즘(푸드 파이트 대회)

2024. 5. 14. 21:22TIL

 

오늘은 간단한 알고리즘 문제를 풀어보자!

 

🔥알고리즘 문제

 

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

 

내가 작성한 코드 

class Solution {
    fun solution(food: IntArray): String {
    var answer: String = ""
    var answerFood= mutableListOf<Int>()
    for( i in 0.. food.size-1){
        answerFood.add(food.map{
            if(it%2!=0) it-1
            else it
        }[i])
    }
    answerFood=answerFood.map{it->it/2}.toMutableList()
    var stringList= mutableListOf<String>()
    for(i in 0.. answerFood.size-1){
        stringList.add(i.toString().repeat(answerFood[i]))
    }
    answer=stringList.joinToString("")
    

    return answer+"0"+answer.reversed()
}
}

 

내가 작성한 코드 해석

food안의 숫자가 짝수여야만하기 때문에  홀수인경우 -1을 해주고  짝수가 된 food요소들을 2로 나눈뒤 answer라는 문자열에 넣는다.  
그 다음 answer  + "0"+answer.reversed()로 상대편의 음식과 나의 음식을 합친 값을 반환해준다

 

다른사람은 어떻게 했을까?

class Solution {
    fun solution(food: IntArray): String {
        val sb = StringBuilder()
        for (i in 1..food.lastIndex) {
            repeat(food[i] / 2) { sb.append(i) }
        }
        val reversed = sb.reversed()
        sb.append(0)
        sb.append(reversed)
        return sb.toString()
    }
}

 

다른 사람들이 작성한 코드 해석

repeat(food[i] / 2) { sb.append(i) }를 통해서 각 요소의 2로 나눈 몫을 string에 추가해서 sb라는 문자열에 넣어준 코드이다. 
내가 작성한 코드와 비교하면 굳이 1을 빼고 2를 나눌필요없이, 2를 나눈 몫을 반환함으로서  해결을 한것을 볼 수 있다

헉! 다른사람은 이렇게 쉽게....

 

더 개선해야할 Point

알고리즘 설계에 있어서 더 생각해보고 작성을 해야겠다

 

 알고리즘 GitHub : https://github.com/kotlin2024/algorithm/commit/e8d8601367f133cc87c0add8aa2b419a60ef3df4

 

'TIL' 카테고리의 다른 글

간단한 API를 직접 만들어보자(2)  (0) 2024.05.16
간단한 API를 직접 만들어보자(1)  (0) 2024.05.15
예외처리를 전역적으로  (0) 2024.05.13
RequestMapping 너는 내가 담당한다  (0) 2024.05.12
@Bean  (0) 2024.05.11