🔗 문제 링크
프로그래머스
SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프
programmers.co.kr
📝 문제 설명
게임개발자인 "죠르디"는 크레인 인형뽑기 기계를 모바일 게임으로 만들려고 합니다.
"죠르디"는 게임의 재미를 높이기 위해 화면 구성과 규칙을 다음과 같이 게임 로직에 반영하려고 합니다.
게임 화면은 "1 x 1" 크기의 칸들로 이루어진 "N x N" 크기의 정사각 격자이며 위쪽에는 크레인이 있고 오른쪽에는 바구니가 있습니다. (위 그림은 "5 x 5" 크기의 예시입니다). 각 격자 칸에는 다양한 인형이 들어 있으며 인형이 없는 칸은 빈칸입니다. 모든 인형은 "1 x 1" 크기의 격자 한 칸을 차지하며 격자의 가장 아래 칸부터 차곡차곡 쌓여 있습니다. 게임 사용자는 크레인을 좌우로 움직여서 멈춘 위치에서 가장 위에 있는 인형을 집어 올릴 수 있습니다. 집어 올린 인형은 바구니에 쌓이게 되는 데, 이때 바구니의 가장 아래 칸부터 인형이 순서대로 쌓이게 됩니다.
만약 같은 모양의 인형 두 개가 바구니에 연속해서 쌓이게 되면 두 인형은 터뜨려지면서 바구니에서 사라지게 됩니다.
크레인 작동 시 인형이 집어지지 않는 경우는 없으나 만약 인형이 없는 곳에서 크레인을 작동시키는 경우에는 아무런 일도 일어나지 않습니다. 또한 바구니는 모든 인형이 들어갈 수 있을 만큼 충분히 크다고 가정합니다.
게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.
⚠️ 제한 사항
- board 배열은 2차원 배열로 크기는 "5 x 5" 이상 "30 x 30" 이하입니다.
- board의 각 칸에는 0 이상 100 이하인 정수가 담겨있습니다.
- 0은 빈 칸을 나타냅니다.
- 1 ~ 100의 각 숫자는 각기 다른 인형의 모양을 의미하며 같은 숫자는 같은 모양의 인형을 나타냅니다.
- moves 배열의 크기는 1 이상 1,000 이하입니다.
- moves 배열 각 원소들의 값은 1 이상이며 board 배열의 가로 크기 이하인 자연수입니다.
✨ 내 코드
import java.util.Arrays;
import java.util.Stack;
class Solution {
public int solution(int[][] board, int[] moves) {
// 책 힌트: board를 열 기준 스택으로 이루어진 배열로 변경
Stack<Integer>[] dolls = new Stack[board.length];
for(int i = 0; i < dolls.length; i++) {
dolls[i] = new Stack<>();
}
// 열 기준으로 dolls 채우기
for (int i = board.length - 1; i >= 0; i--) {
for (int j = 0; j < board.length; j++) {
if(board[i][j] != 0) {
dolls[j].push(board[i][j]);
}
}
}
int count = 0;
Stack<Integer> picked = new Stack<>(); // 뽑은 인형을 담는 바구니
for (int i = 0; i < moves.length; i++) {
int dollIdx = moves[i] - 1; // moves 값은 1부터 시작하므로 -1 처리
// dolls의 특정 인덱스가 비어있다 == 인형이 없다 -> 아무 일도 안 생기고 다음으로
if (dolls[dollIdx].isEmpty()) {
continue;
} else {
int doll = dolls[dollIdx].pop();
// 바구니가 비어있지 않고 가장 최근에 뽑은 인형과 지금 뽑은 인형이 같음
if (!picked.isEmpty() && picked.peek() == doll) {
picked.pop();
count += 2;
} else {
picked.push(doll);
}
}
}
return count;
}
}
// 같은 모양 애니팡
// 없는 곳 집으면 아무 일도 없음
// 터트려져 사라진 인형 개수
💡 풀이
- 입력으로 들어온 board를 열 기준 스택으로 이루어진 배열로 변경한다.
- 반복문을 돌며 움직임을 처리한다
- 1로 만든 배열 속 스택이 비어있다면 인형을 다 뽑았다는 뜻이므로 아무 일도 생기지 않고 넘어간다
- 뽑을 인형을 pop한다.
- 바구니가 비어있지 않고 최근 뽑은 인형과 지금 뽑은 인형이 같으면 바구니 pop하고 count 값을 2 올린다.
- 이외의 경우엔 바구니에 인형을 push한다.
💭 회고
처음엔 각 열에 인형이 있던 마지막 위치를 저장하는 배열을 활용하려 했다.
근데 오히려 그 배열을 관리하는게 너무 헷갈려서 다른 방법을 모색했다.
두번째 방법으로는 board를 열 기준으로 정리한 이차원 배열을 만들었다
근데 이 방법은 인형이 없는 부분이 0이라서 인형을 뽑으려면 반복문을 또 돌아야했다.
결국 괜찮은 방법이 떠오르지 않아 책을 참고했는데 배열에 스택을 넣을 수 있었다..!
스택을 넣으면 바로 pop하면 되니까 너무 편했다.
(아니 근데 이거 아무리 생각해도 레벨1짜리가 아닌거 같다;;)
앞으로 문제를 풀 때는
- 경우에 따라 주어진 배열을 재배열하는 방법을 생각하자
- 스택을 배열 안에 넣을 수 있음을 잊지 말자
'Algorithm > 프로그래머스' 카테고리의 다른 글
| [Java] Lv2. 짝지어 제거하기 (0) | 2025.06.01 |
|---|---|
| [Java] Lv2. 주식가격 (0) | 2025.06.01 |
| [Java] Lv2. 괄호 회전하기 (0) | 2025.06.01 |
| [Java] Lv2. 표 편집 (0) | 2025.06.01 |
| [Java] Lv2. 방문 길이 (1) | 2025.05.25 |