[level 1] 햄버거 만들기(시간초과 이슈...다시 풀기!)

문제

햄버거 가게에서 일을 하는 상수는 햄버거를 포장하는 일을 합니다. 함께 일을 하는 다른 직원들이 햄버거에 들어갈 재료를 조리해 주면 조리된 순서대로 상수의 앞에 아래서부터 위로 쌓이게 되고, 상수는 순서에 맞게 쌓여서 완성된 햄버거를 따로 옮겨 포장을 하게 됩니다. 상수가 일하는 가게는 정해진 순서(아래서부터, 빵 – 야채 – 고기 - 빵)로 쌓인 햄버거만 포장을 합니다. 상수는 손이 굉장히 빠르기 때문에 상수가 포장하는 동안 속 재료가 추가적으로 들어오는 일은 없으며, 재료의 높이는 무시하여 재료가 높이 쌓여서 일이 힘들어지는 경우는 없습니다.

예를 들어, 상수의 앞에 쌓이는 재료의 순서가 [야채, 빵, 빵, 야채, 고기, 빵, 야채, 고기, 빵]일 때, 상수는 여섯 번째 재료가 쌓였을 때, 세 번째 재료부터 여섯 번째 재료를 이용하여 햄버거를 포장하고, 아홉 번째 재료가 쌓였을 때, 두 번째 재료와 일곱 번째 재료부터 아홉 번째 재료를 이용하여 햄버거를 포장합니다. 즉, 2개의 햄버거를 포장하게 됩니다.

상수에게 전해지는 재료의 정보를 나타내는 정수 배열 ingredient가 주어졌을 때, 상수가 포장하는 햄버거의 개수를 return 하도록 solution 함수를 완성하시오.

 

• 1 ≤ ingredient의 길이 ≤ 1,000,000
• ingredient의 원소는 1, 2, 3 중 하나의 값이며, 순서대로 빵, 야채, 고기를 의미합니다

 

문제 풀이 접근법1

1) ingredient의 길이가 4보다 많을 경우(하 4까지는 가능인데...왜 그랬지..^^)에만 반복문을 반복한다.

2) 만약 i가 num보다 작다면 i에서부터 4번 반복하여 해당하는 ingredient의 원소를 pack에 추가한다.

3) 만약 pack이 hambuger와 같다면 answer에 +1을 하고, 해당 원소를 삭제한다. i=0으로 변경하여 다시 반복문을 돌릴 수 있게 한다.

4) 만약 pack이 hambuger와 다르면 탐색을 지속할 수 있도록 i에 +1을 한다.

5) 만약 i가 num보다 크다면 반복문을 탈출한다(더 이상 탐색이 불가하므로)

def solution(ingredient):
    answer = 0
    hambuger = [1, 2, 3, 1]
    pack = []
    i = 0
    
    num = len(ingredient) - 5

    while len(ingredient) > 4:

        if i < num:
            for j in range(i, i+4):
                pack.append(ingredient[j])

            if pack == hambuger:
                answer +=1
                del ingredient[i:i+4]
                i = 0
            else:
                i+=1

            pack.clear()

        else:
            break
    return answer

 

 

문제 풀이 접근법2

1) i가 num이랑 같으면 바로 탈출한다

2) ingredient[i, i+1, i+2, i+3, i+4]가 hambuger와 같으면 answer에 +1을 하고, 해당 원소들을 ingredient에서 삭제한 뒤 i를 0으로 변경한다.

3) 같지 않다면 탐색을 진행하기 위해 i에 +1을 한다. 

def solution(ingredient):
    answer = 0
    hambuger = [1, 2, 3, 1]
    i = 0
    num = len(ingredient) - 4
    
    while len(ingredient) > 4 :

        if i == num:
            break 

        if ingredient[i:i+4] == hambuger:
            answer +=1
            del ingredient[i:i+4]
            i = 0
        else:
            i+=1

   
    return answer

1, 14, 18에서 실패하고

3 ~ 9번 케이스, 11, 12번 케이스에서 시간초과가 발생했다..ㅎ

 

 

문제 풀이 접근법3

1) 반복 조건을 엄청 고민하다가 생각해보니까 현재 시점에서 앞에 남은 원소가 3개가 아니면(즉, 자기 자신 포함 원소가 4개가 아니면) 햄버거를 만들지 못하므로 해당 시점을 반복 조건으로 설정해주면 되겠다는 생각이 들어서 변경했다.

def solution(ingredient):
    answer = 0
    
    hambuger = [1, 2, 3, 1]
    i = 0


    while len(ingredient) - i > 3:

        if ingredient[i:i+4] == hambuger:
            answer +=1
            del ingredient[i:i+4]
            i = 0
        else:
            i+=1

   
    return answer

1, 14, 18번에서 발생했던 실패는 사라졌다.

그런데 시간초과는 여전함...아무리 생각하고 또  생각해보아도 시간초과를 줄일 방법이 생각나지 않아서

결국 답지를 보았어요...

 

 

다른 분들 문제 풀이

1) ingredient의 원소로 for문을 진행한다. 이때, 각 원소들을 package에 추가한다.

2) 만약 package의 뒤에서부터 4개의 원소가 hambuger와 같다면 answer에 +1을 하고, 반복문으로 해당 원소들을 pop하여 꺼낸다

def solution(ingredient):
    answer = 0
    
    hambuger = [1, 2, 3, 1]
    package = []
    
    for i in ingredient:
        package.append(i)
        if package[-4:] == hambuger:
            answer+=1
            for k in range(4):
                package.pop()

   
    return answer

하...그러네...이것도 스택이네...

ingredient에 있는 원소를 새로운 스택에 하나씩 쌓으면서, 새 스택에 쌓여 있는 원소 중 뒤에서부터 4개의 원소가 hambuger와 같다면 햄버거를 만든 것이므로 answer를 +1하고 새 스택에서 해당 원소 4개를 pop한다

진짜갈길이멀다!!!

담번에 비슷한 유형 문제가 나온다면 꼭 기억하기.