[코딩테스트] - 택배 상자 LV2 - ( 프로그래머스(programmers) / java )👍
안녕하세요 나홀로전세집입니다.
자! 오늘의 문제는 택배 상자입니다 !
https://school.programmers.co.kr/learn/courses/30/lessons/131704
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
- 택배상자
영재는 택배상자를 트럭에 싣는 일을 합니다. 영재가 실어야 하는 택배상자는 크기가 모두 같으며 1번 상자부터 n번 상자까지 번호가 증가하는 순서대로 컨테이너 벨트에 일렬로 놓여 영재에게 전달됩니다. 컨테이너 벨트는 한 방향으로만 진행이 가능해서 벨트에 놓인 순서대로(1번 상자부터) 상자를 내릴 수 있습니다. 하지만 컨테이너 벨트에 놓인 순서대로 택배상자를 내려 바로 트럭에 싣게 되면 택배 기사님이 배달하는 순서와 택배상자가 실려 있는 순서가 맞지 않아 배달에 차질이 생깁니다. 따라서 택배 기사님이 미리 알려준 순서에 맞게 영재가 택배상자를 실어야 합니다.
만약 컨테이너 벨트의 맨 앞에 놓인 상자가 현재 트럭에 실어야 하는 순서가 아니라면 그 상자를 트럭에 실을 순서가 될 때까지 잠시 다른 곳에 보관해야 합니다. 하지만 고객의 물건을 함부로 땅에 둘 수 없어 보조 컨테이너 벨트를 추가로 설치하였습니다. 보조 컨테이너 벨트는 앞 뒤로 이동이 가능하지만 입구 외에 다른 면이 막혀 있어서 맨 앞의 상자만 뺄 수 있습니다(즉, 가장 마지막에 보조 컨테이너 벨트에 보관한 상자부터 꺼내게 됩니다). 보조 컨테이너 벨트를 이용해도 기사님이 원하는 순서대로 상자를 싣지 못 한다면, 더 이상 상자를 싣지 않습니다.
예를 들어, 영재가 5개의 상자를 실어야 하며, 택배 기사님이 알려준 순서가 기존의 컨테이너 벨트에 네 번째, 세 번째, 첫 번째, 두 번째, 다섯 번째 놓인 택배상자 순서인 경우, 영재는 우선 첫 번째, 두 번째, 세 번째 상자를 보조 컨테이너 벨트에 보관합니다. 그 후 네 번째 상자를 트럭에 싣고 보조 컨테이너 벨트에서 세 번째 상자 빼서 트럭에싣습니다. 다음으로 첫 번째 상자를 실어야 하지만 보조 컨테이너 벨트에서는 두 번째 상자를, 기존의 컨테이너 벨트에는 다섯 번째 상자를 꺼낼 수 있기 때문에 더이상의 상자는 실을 수 없습니다. 따라서 트럭에는 2개의 상자만 실리게 됩니다.
택배 기사님이 원하는 상자 순서를 나타내는 정수 배열 order가 주어졌을 때, 영재가 몇 개의 상자를 실을 수 있는지 return 하는 solution 함수를 완성하세요.
제한사항
- 1 ≤ order의 길이 ≤ 1,000,000
- order는 1이상 order의 길이 이하의 모든 정수가 한번씩 등장합니다.
- order[i]는 기존의 컨테이너 벨트에 order[i]번째 상자를 i+1번째로 트럭에 실어야 함을 의미합니다.
문제 상황은 영재가 배달원 아저씨가 알려준 순서대로 택배 상자를 넣어야 합니다.
문제를 딱 보자마자 스택을 사용해야 단 것을 눈치챘습니다. 컨베이어 벨트는 일반 변수로 놔두고, 예비 컨베이어 벨트를 스택으로 선언하면 되겠다고 생각하고 문제를 풀기 시작했습니다.
먼저 변수입니다!
int answer = 0; // 트럭에 실리는 상자
int box = 1; // 컨베이어 벨트
Stack<Integer> temp = new Stack<>(); // 예비 컨베이어 벨트
저는 다음과 같은 변수들을 사용했습니다.
트럭에 실릴 때마다 1씩 증가하는 answer
컨베이어 벨트에서 꺼낼 때마다 증가하는 box
스택으로 예비 컨베이어 벨트를 선언했습니다.
while (answer < order.length) {
if (!temp.empty()) {
if (order[answer] < box && order[answer] < temp.peek()) { // 오더 값이 컨베이어보다 작고 예비 컨베이어랑 다를 때
return answer;
}
}
// 컨베이어에 있는게 오더랑 같을 때
if (order[answer] == box) {
box++;
answer++;
continue;
}
if (!temp.empty()) {
if (order[answer] == temp.peek()) { // 예비 컨베이어 제일 뒷쪽 박스가 오더랑 같을 때
temp.pop();
answer++;
continue;
}
}
// 컨베이어에 있는 박스가 예비 컨베이어에 올라갈 때
temp.push(box);
box++;
}
다음은 알고리즘입니다.
사실, 문제만 이해하면 코드는 간단하게 짜집니다.
스택을 사용해서 문제를 풀어야 한다고 생각하는 것이 가장 어려웠던 문제였습니다.
전체 코드
import java.util.Stack;
class Solution { // 택배 상자
public int solution(int[] order) {
int answer = 0; // 트럭에 실리는 상자
int box = 1; // 컨베이어 벨트
Stack<Integer> temp = new Stack<>(); // 예비 컨베이어 벨트
while (answer < order.length) {
if (!temp.empty()) {
if (order[answer] < box && order[answer] < temp.peek()) { // 오더 값이 컨베이어보다 작고 예비 컨베이어랑 다를 때
return answer;
}
}
if (order[answer] == box) {
box++;
answer++;
continue;
}
if (!temp.empty()) {
if (order[answer] == temp.peek()) { // 예비 컨베이어 제일 뒷쪽 박스가 오더랑 같을 때
temp.pop();
answer++;
continue;
}
}
// 컨베이어에 있는 박스가 예비 컨베이어에 올라갈 때
temp.push(box);
box++;
}
return answer;
}
}
오늘도 즐코딩 하시고 좋은 하루 되세요~