题目地址:
https://www.lintcode.com/problem/zombie-in-matrix/description
给定一个 m m m行 n n n列的二维矩阵 A A A, 2 2 2代表墙, 1 1 1代表僵尸, 0 0 0代表人。每一步后,僵尸可以将其紧邻的人变为僵尸,但不能对墙产生任何影响。问多少步后所有的人都变为僵尸。如果总存在人变不成僵尸,则返回 − 1 -1 −1。
思路是BFS。先将所有僵尸入队,然后每步感染一下僵尸周围的人,并将那些新感染的人入队。最后check一下是否已经不存在人了。代码如下:
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
/**
* @param grid: a 2D integer grid
* @return: an integer
*/
public int zombie(int[][] grid) {
// write your code here
if (grid == null || grid.length == 0 || grid[0].length == 0) {
return -1;
}
// 四个方向
int[][] dirs = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
int pCount = 0;
Queue<int[]> queue = new LinkedList<>();
// 将所有僵尸入队
for (int i = 0; i < grid.length; i++) {
for (int j = 0; j < grid[0].length; j++) {
if (grid[i][j] == 1) {
queue.offer(new int[]{i, j});
} else if (grid[i][j] == 0) {
pCount++;
}
}
}
// 记录感染的步数
int step = 0;
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
int[] cur = queue.poll();
// 如果cur紧邻的位置有人,则将其感染为僵尸并入队
for (int j = 0; j < 4; j++) {
int x = cur[0] + dirs[j][0];
int y = cur[1] + dirs[j][1];
if (inBound(grid, x, y) && grid[x][y] == 0) {
grid[x][y] = 1;
// 感染了一个人
pCount--;
queue.offer(new int[]{x, y});
}
}
}
// 如果有新被感染的人,说明需要感染的次数需要 + 1,所以令step++
if (!queue.isEmpty()) {
step++;
}
}
// 如果pCount等于0,说明所有人都被感染了,返回step;否则返回-1
return pCount == 0 ? step : -1;
}
private boolean inBound(int[][] grid, int x, int y) {
return 0 <= x && x < grid.length && 0 <= y && y < grid[0].length;
}
}
时空复杂度 O ( m n ) O(mn) O(mn)。