传送门:723. Candy Crush
Problem:
This question is about implementing a basic elimination algorithm for Candy Crush. Given a 2D integer array board representing the grid of candy, different positive integers board[i][j] represent different types of candies. A value of board[i][j] = 0 represents that the cell at position (i, j) is empty. The given board represents the state of the game following the player’s move. Now, you need to restore the board to a stable state by crushing candies according to the following rules:
You need to perform the above rules until the board becomes stable, then return the current board.
Example 1:
Input: [[110,5,112,113,114],[210,211,5,213,214],[310,311,3,313,314],[410,411,412,5,414],[5,1,512,3,3],[610,4,1,613,614],[710,1,2,713,714],[810,1,2,1,1],[1,1,2,2,2],[4,1,4,4,1014]] Output: [[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[110,0,0,0,114],[210,0,0,0,214],[310,0,0,113,314],[410,0,0,213,414],[610,211,112,313,614],[710,311,412,613,714],[810,411,512,713,1014]] Explanation:
Note:
The length of board will be in the range [3, 50].
The length of board[i] will be in the range [3, 50].
Each board[i][j] will initially start as an integer in the range [1, 2000].
思路: 这不就是开心消消乐么,思路很直接,只要根据列和行检测出连续三个以上的块后,记录下来,消去,并继续更新,直到找不到连续三个块为止。
代码如下:
public int[][] candyCrush(int[][] board) {
int n = board.length;
int m = board[0].length;
boolean[][] next = new boolean[n][m];
while (getNext(next, board)){
List<Integer>[] ans = new ArrayList[m];
for (int i = 0; i < m; ++i){
ans[i] = new ArrayList<>();
for (int j = n - 1; j >= 0; --j){
if (!next[j][i]) ans[i].add(board[j][i]);
}
}
board = new int[n][m];
// 重写
for (int j = 0; j < m; ++j){
for (int i = 0; i < ans[j].size(); ++i){
board[n - 1 - i][j] = ans[j].get(i);
}
}
next = new boolean[n][m];
}
return board;
}
public boolean getNext(boolean[][] next, int[][] board){
boolean exist = false;
// 遍历行
int n = board.length;
int m = board[0].length;
for (int i = 0; i < n; ++i){
int prev = board[i][0];
int count = 1;
for (int j = 1; j < m; ++j){
if (board[i][j] == prev && board[i][j] != 0){
count ++;
}
else{
if (count >= 3){
for (int l = 0; l < count; ++l){
next[i][j - 1 - l] = true;
}
exist = true;
}
count = 1;
}
prev = board[i][j];
}
if (count >= 3){
for (int l = 0; l < count; ++l){
next[i][m - 1 - l] = true;
}
exist = true;
}
}
// 遍历列
for (int i = 0; i < m; ++i){
int prev = board[0][i];
int count = 1;
for (int j = 1; j < n; ++j){
if (board[j][i] == prev && board[j][i] != 0){
count ++;
}
else{
if (count >= 3){
for (int l = 0; l < count; ++l){
next[j - 1 - l][i] = true;
}
exist = true;
}
count = 1;
}
prev = board[j][i];
}
if (count >= 3){
for (int l = 0; l < count; ++l){
next[n - 1 - l][i] = true;
}
exist = true;
}
}
return exist;
}