문제
문제 설명
영재는 택배상자를 트럭에 싣는 일을 합니다. 영재가 실어야 하는 택배상자는 크기가 모두 같으며 1번 상자부터 n번 상자까지 번호가 증가하는 순서대로 컨테이너 벨트에 일렬로 놓여 영재에게 전달됩니다. 컨테이너 벨트는 한 방향으로만 진행이 가능해서 벨트에 놓인 순서대로(1번 상자부터) 상자를 내릴 수 있습니다. 하지만 컨테이너 벨트에 놓인 순서대로 택배상자를 내려 바로 트럭에 싣게 되면 택배 기사님이 배달하는 순서와 택배상자가 실려 있는 순서가 맞지 않아 배달에 차질이 생깁니다. 따라서 택배 기사님이 미리 알려준 순서에 맞게 영재가 택배상자를 실어야 합니다.
만약 컨테이너 벨트의 맨 앞에 놓인 상자가 현재 트럭에 실어야 하는 순서가 아니라면 그 상자를 트럭에 실을 순서가 될 때까지 잠시 다른 곳에 보관해야 합니다. 하지만 고객의 물건을 함부로 땅에 둘 수 없어 보조 컨테이너 벨트를 추가로 설치하였습니다. 보조 컨테이너 벨트는 앞 뒤로 이동이 가능하지만 입구 외에 다른 면이 막혀 있어서 맨 앞의 상자만 뺄 수 있습니다(즉, 가장 마지막에 보조 컨테이너 벨트에 보관한 상자부터 꺼내게 됩니다). 보조 컨테이너 벨트를 이용해도 기사님이 원하는 순서대로 상자를 싣지 못 한다면, 더 이상 상자를 싣지 않습니다.
예를 들어, 영재가 5개의 상자를 실어야 하며, 택배 기사님이 알려준 순서가 기존의 컨테이너 벨트에 네 번째, 세 번째, 첫 번째, 두 번째, 다섯 번째 놓인 택배상자 순서인 경우, 영재는 우선 첫 번째, 두 번째, 세 번째 상자를 보조 컨테이너 벨트에 보관합니다. 그 후 네 번째 상자를 트럭에 싣고 보조 컨테이너 벨트에서 세 번째 상자 빼서 트럭에싣습니다. 다음으로 첫 번째 상자를 실어야 하지만 보조 컨테이너 벨트에서는 두 번째 상자를, 기존의 컨테이너 벨트에는 다섯 번째 상자를 꺼낼 수 있기 때문에 더이상의 상자는 실을 수 없습니다. 따라서 트럭에는 2개의 상자만 실리게 됩니다.
택배 기사님이 원하는 상자 순서를 나타내는 정수 배열 order
가 주어졌을 때, 영재가 몇 개의 상자를 실을 수 있는지 return 하는 solution 함수를 완성하세요.
제한사항
- 1 ≤ order의 길이 ≤ 1,000,000
- order는 1이상 order의 길이 이하의 모든 정수가 한번씩 등장합니다.
- order[i]는 기존의 컨테이너 벨트에 order[i]번째 상자를 i+1번째로 트럭에 실어야 함을 의미합니다.
입출력 예
order | result |
---|---|
[4, 3, 1, 2, 5] | 2 |
[5, 4, 3, 2, 1] | 5 |
코드
function solution(order) {let answer = 0;let idx = 0;const tempBelt = [];for (let box = 1; box <= order.length; box++) {if (box !== order[idx]) {tempBelt.push(box);continue;}answer++;idx++;while (tempBelt.length && tempBelt.at(-1) === order[idx]) {tempBelt.pop();answer++;idx++;}}return answer;}
문제 풀이
stack
을 이용하는 문제였다. 처음에는 메인 벨트와 보조 벨트를 모두 배열로 선언해서 push
와 shift
연산을 해서 문제를 해결하려고 했지만, 시간 복잡도를 고려했을 때 좋지 않은 접근 방법이었다. 다른 사람의 풀이를 참고해서 풀게 되었지만, idx 변수를 두어서 order와 일치하는지를 계속 비교해주는 방법으로 해결하였다.
먼저 for 문으로 순회하면서 order에 있는 값과 비교를 하고 일치하지 않는다면 tempBelt에 push한다.
for (let box = 1; box <= order.length; box++) {if (box !== order[idx]) {tempBelt.push(box);continue;}...}
그 다음 줄 부터는 order와 현재 box가 일치한다는 것이므로 idx와 answer를 1씩 증가 시켜준다. 그리고 tempBelt를 확인해서 일치하는지를 한 번 더 확인해준다. tempBelt는 stack
구조이므로 마지막 요소를 비교해야 한다. 따라서 자바스크립트의 at()
메서드를 사용해서 비교한다.
// 위 if 문에서 걸러지지 않았으므로 일치하는 것이다.answer++;idx++;// tempBelt가 비어있지 않고// tempBelt의 제일 마지막에 있는 상자의 번호와 order의 상자가 일치할 때.while (tempBelt.length && tempBelt.at(-1) === order[idx]) {tempBelt.pop();answer++;idx++;}