Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 | 31 |
Tags
- 세그먼트 트리
- swea
- 프로그래머스
- 시뮬레이션
- 투 포인터
- mst
- 백준
- Kotlin
- Network
- Effective Java
- JUnit 5
- 알고리즘
- java
- 유니온 파인드
- 위상정렬
- 플로이드-와샬
- 수학
- 에라토스테네스의 체
- 이분탐색
- 문자열
- 후니의 쉽게 쓴 시스코 네트워킹
- 동적계획법
- dfs
- 구현
- 그리디
- CS
- 스택
- 백트래킹
- 완전탐색
- BFS
Archives
반갑습니다!
[프로그래머스] 크레인 인형뽑기 게임 본문
풀이
바구니를 stack
으로 구현하면 되는 문제이다. 인형을 집고 떨어뜨리기 전에 stack
의 가장 위에 있는 인형의 종류를 확인하여 같은 인형이면 pop
해주면 된다.
코드
C++
#include <string>
#include <vector>
#include <stack>
using namespace std;
int solution(vector<vector<int>> board, vector<int> moves) {
int answer = 0;
stack<int> s;
for(int i=0; i<moves.size(); i++){
int pick = moves[i] - 1;
int h = 0;
while(h < board.size() && board[h][pick] == 0) h++;
if(h < board.size() && board[h][pick] != 0){
if(!s.empty() && s.top() == board[h][pick]) {
answer += 2;
s.pop();
}
else s.push(board[h][pick]);
board[h][pick] = 0;
}
}
return answer;
}
Java
import java.util.Stack;
class Solution {
public int solution(int[][] board, int[] moves) {
int answer = 0;
int n = board.length;
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < moves.length; i++) {
int x = moves[i] - 1;
for (int y = 0; y < n; y++) {
if (board[y][x] != 0) {
int doll = board[y][x];
board[y][x] = 0;
if (!stack.empty() && stack.peek() == doll) {
stack.pop();
answer += 2;
} else {
stack.push(doll);
}
break;
}
}
}
return answer;
}
}
'알고리즘 문제 풀이' 카테고리의 다른 글
[프로그래머스] 카카오 프렌즈 컬러링 북 (0) | 2020.04.03 |
---|---|
[프로그래머스] 멀쩡한 사각형 (0) | 2020.04.03 |
[프로그래머스] 문자열 압축 (0) | 2020.04.03 |
[프로그래머스] 기능개발 (0) | 2020.04.02 |
[ 프로그래머스] 다리를 지나는 트럭 (0) | 2020.04.02 |