【Lintcode】598. Zombie in Matrix

题目地址：

https://www.lintcode.com/problem/zombie-in-matrix/description

给定一个$m$行$n$列的二维矩阵$A$，$2$代表墙，$1$代表僵尸，$0$代表人。每一步后，僵尸可以将其紧邻的人变为僵尸，但不能对墙产生任何影响。问多少步后所有的人都变为僵尸。如果总存在人变不成僵尸，则返回$-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(mn)$。