[swift] 프로그래머스 - 햄버거 만들기

프로그래머스

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

햄버거 재료 배열이 주어집니다.

배열 중 연속적으로 [빵, 야채, 고기, 빵]이 나온다면

햄버거가 될 수 있습니다.

 

1: 빵, 2: 야채, 3: 고기입니다.

 

[2, 1, 1, 2, 3, 1, 2, 3, 1]의 배열이 주어질 때

3번째 인덱스부터 [1,2,3,1]이 되기 때문에 햄버거가 만들어지며

해당 범위를 지우게 된다면 [2,1,2,3,1]이 남습니다.

다시 남은 배열에서 2번째 인덱스부터

햄버거를 만들 수 있는 [1,2,3,1]이 만들어져서

2개의 햄버거를 만들 수 있습니다.

 

---

 

나의 풀이

import Foundation

func solution(_ ingredient:[Int]) -> Int {
    var burger = ingredient
    var count = 0
    var start = 0
    while burger.count >= 4 {
        let temp = count
        for i in start..<burger.count-3 {
            if burger[i...i+3] == [1,2,3,1] {
                burger[i...i+3] = []
                count += 1
                if burger.count <= 6 { break }
                start = i-2
                break
            }
        }
        if temp == count { break }
    }
    return count
}

 

저는 4개의 공간을 가진 배열을 만들어

burger재료를 처음부터 끝까지 [1,2,3,1]에 해당하는지 탐색했습니다.

만약 [1,2,3,1]에 해당되면, 해당 인덱스는 지워주고 count += 1을 했습니다.

그리고 처음부터 다시 [1,2,3,1]을 탐색했습니다.

 

하지만 해당 조건만으로는

시간제한에 걸려 통과하지 못했습니다.

 

그래서 햄버거 배열을 뺀 후

처음부터 다시 탐색하지 않고

 

빠진 index에서 2칸만 뒤로 가 다시 탐색했습니다.

최소 2칸을 뒤로가야 완벽하게  다시 햄버거를 찾을 수 있었습니다.

 

인덱스가 음수로 나오는걸 방지하기 위해

burger의 카운트가 6보다 작을 때는

인덱스를 뒤로보내 탐색하지 않았습니다.

(만약 5의 count를 가지고 있을 때

burger[1... 4]를 검사하고 start값인 1에 -2를 하면 음수가 나와 에러가 발생합니다)

 

미리 count값을 복사해 놓은 temp값이

burger배열을 탐색하고도 같다면

더 이상 햄버거를 만들 수 있는 배열이 없다는 것이기 때문에

while문을 나옵니다.

 

---

 

참고할만한 풀이

import Foundation

func solution(_ ingredient:[Int]) -> Int {
    var stacks: [Int] = []
    var count: Int = 0
    for ingredient in ingredient {
        stacks.append(ingredient)
        let suffix = stacks.suffix(4)
        if suffix == [1,2,3,1] {
            count += 1
            stacks.removeLast(4)
        }
    }
    return count
}

 

.suffix(4)는

뒤에서부터 4개의 인덱스를 불러올 수 있습니다.

 

suffix() 함수를 썼을 때 좋은 점이

배열에 2개의 값 밖에 없다 해도

suffix(4)는 에러가 나지 않고

2개의 값만 가져옵니다.

 

따라서 인덱스를 활용했을 때는

에러를 처리해야 하는 조건이 필요했는데

suffix()를 쓰면 불필요하기 때문에 더 깔끔한 코드를 만들 수 있어 보입니다.

 

---