【Lintcode】199. Judge Connection

题目地址：

https://www.lintcode.com/problem/judge-connection/description

给定一个二维矩阵，再给定一个数$k$，问矩阵中所有值为$k$的点是否都彼此连通。

可以用DFS。第一次遇到$k$的时候将其代表的连通块全变为$k+1$（选取任意一个单射函数都可以），并记录已经染色（这个类似于flood-fill），接下来如果再次遇到了$k$就说明答案是false，提前返回；否则最终返回true即可。代码如下：

public class Solution {
    /**
     * @param arr: the arr
     * @param k:   the k
     * @return: if all instances of value k in arr are connected
     */
    public boolean judgeConnection(int[][] arr, int k) {
        // Write your code here.
        if (arr == null || arr.length == 0 || arr[0].length == 0) {
            return true;
        }
        
        boolean changed = false;
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[0].length; j++) {
                if (arr[i][j] == k) {
                    if (changed) {
                        return false;
                    }
                    
                    dfs(i, j, arr, k);
                    changed = true;
                }
            }
        }
        
        return true;
    }
    
    private void dfs(int x, int y, int[][] arr, int k) {
        if (!inBound(x, y, arr) || arr[x][y] != k) {
            return;
        }
        
        // fill成k + 1
        arr[x][y] = k + 1;
        dfs(x + 1, y, arr, k);
        dfs(x - 1, y, arr, k);
        dfs(x, y + 1, arr, k);
        dfs(x, y - 1, arr, k);
    }
    
    private boolean inBound(int x, int y, int[][] arr) {
        return 0 <= x && x < arr.length && 0 <= y && y < arr[0].length;
    }
}

时空复杂度$O(mn)$。