leetcode 刷题第三天

934 最短的桥

给你一个大小为 n x n 的二元矩阵 grid ，其中 1 表示陆地，0 表示水域。

岛 是由四面相连的 1 形成的一个最大组，即不会与非组内的任何其他 1 相连。grid 中 恰好存在两座岛 。

你可以将任意数量的 0 变为 1 ，以使两座岛连接起来，变成 一座岛 。

返回必须翻转的 0 的最小数目。

实例1：

输入：grid = [[0,1],[1,0]]
输出：1

实例2：

输入：grid = [[0,1,0],[0,0,0],[0,0,1]]
输出：2

实例3：

输入：grid = [[1,1,1,1,1],[1,0,0,0,1],[1,0,1,0,1],[1,0,0,0,1],[1,1,1,1,1]]
输出：1

代码：（一开始题都看不明白，难受，运用dfs+bfs搜索路径） 

class Solution {
public:
    int shortestBridge(vector<vector<int>>& grid) {
         m = grid.size(), n = grid[0].size();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (grid[i][j] == 1) {
                    // dfs 将所有源节点入队列
                    dfs(grid, i, j);
                    // bfs 搜索最短路径
                    return bfs(grid);
                }
            }
        }
        return 0;
    }
private:
int m, n;
    int dx[4] = {1, -1, 0, 0};
    int dy[4] = {0, 0, 1, -1};
    queue<pair<int, int>> q;
    void dfs(vector<vector<int>> &grid, int i, int j) {
        if (i < 0 || j < 0 || i >= m || j >= m) {
            return;
        } 
        if (grid[i][j] != 1) {
            return;
        }
        // 访问过的节点设为 -1，防止再次访问
        grid[i][j] = -1;
        q.push({i, j});
        dfs(grid, i + 1, j);
        dfs(grid, i - 1, j);
        dfs(grid, i, j + 1);
        dfs(grid, i, j - 1);
    }
    
    int bfs(vector<vector<int>> &grid) {
        int res = 0;
        while(!q.empty()) {
            int qSize = q.size();
            for (int k = 0; k < qSize; k++) {
                auto p = q.front();
                q.pop();
                int x = p.first, y = p.second;
                for (int r = 0; r < 4; r++) {
                    int nx = x + dx[r], ny = y + dy[r]; 
                    if (nx < 0 || ny < 0 || nx >= m || ny >= n) {
                        continue;
                    }
                    if (grid[nx][ny] == -1) {
                        continue;
                    }
                    if (grid[nx][ny] == 1) {
                        return res;
                    } 
                    grid[nx][ny] = -1;
                    q.push({nx, ny});
                }        
            }
            res++;
        }
        return res;
    }
};