수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다. 대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다. 중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.
이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다. 또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다. 이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.
예를 들어, 3가지의 음식이 준비되어 있으며, 칼로리가 적은 순서대로 1번 음식을 3개, 2번 음식을 4개, 3번 음식을 6개 준비했으며, 물을 편의상 0번 음식이라고 칭한다면, 두 선수는 1번 음식 1개, 2번 음식 2개, 3번 음식 3개씩을 먹게 되므로 음식의 배치는 "1223330333221"이 됩니다. 따라서 1번 음식 1개는 대회에 사용하지 못합니다.
수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.
일단은 문제 자체를 이해하는 게 중요하다. 함수의 입력으로 정수 배열을 받게 되는데, ( [1, 3, 4, 6] 과 같은 모양의) 이 정수의 배열의 뜻은 이렇다.
food[i]
는 i번 음식의 수입니다.
food[0]
은 수웅이가 준비한 물의 양이며, 항상 1입니다.
정답의 길이가 3 이상인 경우만 입력으로 주어집니다.
입력이 [1, 3, 4, 6] 이면, 1번 음식 3개 2번 음식 4개 3번 음식 6개를 준비했으니, 음식들을 각각 반으로 나눠서 양쪽에 1번 음식 1개, 2번 음식 2개, 3번음식 3개... 이런식으로 나눠야 한다는 것이고 답은 "1223330333221"이 되어야 한다.
내가 작성한 코드는 아래와 같다.
import Foundation
func solution(_ food:[Int]) -> String {
var foods = "", temp = 0, result = ""
for i in food {
for _ in 0..<i/2 {
foods += "\(temp)"
}
temp += 1
}
result = "\(foods)" + "0" + "\(String(foods.reversed()))"
return result
}
첫 번째로 할 일은 입력으로 받은 배열을 해석해서 음식을 어떻게 테이블에 올려놓을지 결정하는 것
대칭으로 이루어져야 할 결과 특성을 생각해서, 일단은 좌측 테이블 배열을 foods 변수에 저장하기로 했다.
입력으로 배열의 원소를 받고, 그 원소를 나누기 2하면 소수점은 버리고 정수부분만 결과로 사용하기 때문에(Int 자료형을 사용하기 때문) 중첩반복문을 사용해서 foods 문자열에 준비된 음식의 반을 저장했다.
마지막으로는 result 변수에 foods 문자열을 저장하고, 중간이 될 부분에 물을 의미하는 0을 저장, foods를 거꾸로 한 문자열을 모두 더해주면, 결과가 완성되었다.
출력은 아래와 같이 나왔다.
print(solution([1,3,4,6]))
// 음식 1 -> 3개, 음식 2 -> 4개, 음식 3 -> 6개
출력 : 1223330333221
문제를 푸는데 가장 도움이 되었던 것은 "문자열은 서로 더할 수 있다"는 포인트였다.
foods += "\(temp)"
구문에서 볼 수 있듯이 Swift의 문자열 특성을 이용하면 쉽게 문자열에 요소를 더할 수 있었음
두 번째로 중요한 포인트는 String 자료형의 reversed() 메소드였다.
.reversed()
메소드는 ReversedCollection<String>
자료형을 반환하는데, 이건 기본적으로 콜렉션 자료형이라서 String
으로 형변환을 통해서 문자열에 더해줄 수 있었다.
천천히 어떻게 문제를 해결할지 고민하면 다 풀린다... 조금 오래걸려도 말이지
오래 고민하는 지금의 시간이 나중에는 많은 도움이 되리라고 생각하면서 열공해야겠다!