문제 설명
햄버거 가게에서 일을 하는 상수는 햄버거를 포장하는 일을 합니다. 함께 일을 하는 다른 직원들이 햄버거에 들어갈 재료를 조리해 주면 조리된 순서대로 상수의 앞에 아래서부터 위로 쌓이게 되고, 상수는 순서에 맞게 쌓여서 완성된 햄버거를 따로 옮겨 포장을 하게 됩니다. 상수가 일하는 가게는 정해진 순서(아래서부터, 빵 – 야채 – 고기 - 빵)로 쌓인 햄버거만 포장을 합니다. 상수는 손이 굉장히 빠르기 때문에 상수가 포장하는 동안 속 재료가 추가적으로 들어오는 일은 없으며, 재료의 높이는 무시하여 재료가 높이 쌓여서 일이 힘들어지는 경우는 없습니다.
예를 들어, 상수의 앞에 쌓이는 재료의 순서가 [야채, 빵, 빵, 야채, 고기, 빵, 야채, 고기, 빵]일 때, 상수는 여섯 번째 재료가 쌓였을 때, 세 번째 재료부터 여섯 번째 재료를 이용하여 햄버거를 포장하고, 아홉 번째 재료가 쌓였을 때, 두 번째 재료와 일곱 번째 재료부터 아홉 번째 재료를 이용하여 햄버거를 포장합니다. 즉, 2개의 햄버거를 포장하게 됩니다.
상수에게 전해지는 재료의 정보를 나타내는 정수 배열 ingredient가 주어졌을 때, 상수가 포장하는 햄버거의 개수를 return 하도록 solution 함수를 완성하시오.
제한사항
- 1 ≤ ingredient의 길이 ≤ 1,000,000
- ingredient의 원소는 1, 2, 3 중 하나의 값이며, 순서대로 빵, 야채, 고기를 의미합니다.
내가 푼 풀이
처음에는 배열을 순회해서 빵(1) 을 만나면 빵부터 index+3 까지 빼서 햄버거가 완성되는지 확인을 했다.
햄버거가 만들어지면 카운트하고 배열에서 뺴고 인덱스를 다시 0부터 시작해서 순회했다.
이 방법으로 시도를 했지만 시간초과가 떴다
배열의 최대 크기가 1,000,000 이고, 햄버거가 만들어지려면 정해진 순서가 있다.
위의 방법은 대충 짐작하면 시간복잡도 O(N^2) 가 걸리고 이보다 더 빠른방법을 이용해야한다.
또한 빵이 아닌 재료들이 지나가도 재사용이 가능하다는 점을 이용해야 해서 스택(Stack)을 이용하려고 한다.
Swift 에서 스택을 구현하려면 직접 코딩을 해야하지만 스택의 FILO만 이용하기로 한다
1. 재료배열을 순회해서 스택에 넣는다.
2. 스택의 양이 4 이상이면, 스택의 마지막인덱스로부터 4 범위의 배열을 검사한다.
3. 검사한 배열이 [1,2,3,1] 인 햄버거 만드는조건을 충족하는경우, popLast() ( O(1) ) 를 이용해서 스택에 쌓인 재료를 빼낸다.
4. 만들어진 햄버거 수 출력한다.
Swift의 배열은 첫번째 원소를 뺄 땐 배열의 모든 원소를 순회해야하지만, 마지막 원소를 뺄땐 마지막 원소만 접근한다.
이는 배열의 특징인 연속적으로 원소가 존재하기 때문이다.
임의의 위치에 원소를 추가하는 경우, 삽입되는 위치 이후의 원소들을 한칸씩 뒤로 밀어내야 하고
임의의 위치에 원소를 삭제하는 경우, 삭제되는 위치 이후의 원소들을 한칸씩 앞당겨야 하기 때문이다.
최악의 경우 O(N) 만큼 걸린다.
또한 contains 역시 배열 원소 값 하나하나 체크해야하므로 최악의 경우 O(N) 만큼 걸린다.
import Foundation
func solution(_ ingredient:[Int]) -> Int {
var stack = [Int]()
var result = 0
var check = [1,2,3,1]
for i in 0..<ingredient.count { // 재료 순회
stack.append(ingredient[i])
if stack.count >= 4 { // 스택에 재료 4개가 쌓이면
var arr = [Int]()
for i in stack.endIndex-4..<stack.endIndex { // 스택에 맨위로부터 4개 추출
arr.append(stack[i])
}
if arr == check { // 추출한 재료가 햄버거만드는 조건을 충족하는 경우
result += 1
for i in 0..<4 {
stack.popLast() //재료 빼기
}
}
}
}
return result
}
'코딩테스트 > 프로그래머스' 카테고리의 다른 글
콜라 문제 Swift (0) | 2023.04.10 |
---|---|
옹알이(2) Swift (0) | 2023.04.10 |
푸드 파이트 대회 Swift (0) | 2023.04.10 |
과일 장수 Swift (0) | 2023.04.10 |
기사단원의 무기 Swift (0) | 2023.04.10 |