새소식

알고리즘 테스트 ⏲/프로그래머스

[프로그래머스] 햄버거 만들기 풀이 / JavaScript

  • -
 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 설명

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

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

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


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

 

입출력 예
[2, 1, 1, 2, 3, 1, 2, 3, 1] 2
[1, 3, 2, 1, 2, 1, 3, 1, 2] 0

 

 

내 풀이 (1차시도 / 타임아웃)

function solution(ingredient) {
  let cnt = 0;
  ingredient = ingredient.join("");
  while (true) {
    if (ingredient.includes("1231")) {
      ingredient = ingredient.replace("1231", "");
      cnt += 1;
    } else return cnt;
  }
}

실패 케이스는 없으나 타임아웃이 여러 케이스에서 발생했습니다. 성공은 했으나 제한시간에 아슬아슬하게 통과한 케이스도 있어서 사실 그런 케이스들도 성공이라고 보기는 어렵죠. 

 

원인은 include와 replace에서 발생했을겁니다. ingredient의 최대 길이가 1000000이기 때문에 while문안에서 문자열 탐색까지 들어가는 복잡도가 최악의 경우 n^2 이상이 되니 시간이 초과될 것을 고려하지 못했습니다. 

 

 

내 풀이 (2차시도 / 통과)

function solution(ingredient) {
    let cnt = 0;
    for (let i = 3; i < ingredient.length; i += 1) {
        const now = ingredient[i];
        if (now === 1 && ingredient[i - 1] === 3 && ingredient[i - 2] === 2 && ingredient[i - 3] === 1) {
            ingredient.splice(i - 3, 4);
            cnt += 1;
            i -= 4;
        }
    }
    return cnt;
}

 

ingredient를 무조건 한번만 순회하는 것으로 알고리즘을 변경하고 현재 1이 위치한 경우 이전 인덱스를 -3까지 확인하여 햄버거가 완성된 경우, 그만큼 요소를 제거해주고 제거해준 만큼 인덱스를 다시 되돌려줍니다. 그러면 원래 위치로 돌아가 햄버거를 다시 확인하는 방식입니다.

 

이 시도는 주어진 테스트케이스의 경우 반드시 2초내에 성공하지만 더 적합한 자료구조를 사용한다면 통과 속도를 높일 수 있겠습니다.

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.