Leetcode934-最短的桥

博客介绍了使用多源BFS算法求矩阵中最短桥距离的方法。先遍历矩阵,找到首个值为1的点,对其进行BFS,将所属连通块边界的1装入队列,再对队列元素BFS,首次遇到其他连通块的1时返回距离。时间和空间复杂度均为O(n*m)。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

方法:多源bfs

        先遍历矩阵,找到第一个位置为1的点,对其进行bfs,把他所属的连通块的边界1都装入队列中。随后对队列中的元素进行bfs,首先遇到其他联通块的1即表示最短的桥的距离,返回答案即可。

class Solution {
public:
    int n,m;
    int dx[4]={0,0,1,-1};
    int dy[4]={1,-1,0,0};
    queue<pair<int,int>> q;
    void bfs(int i,int j,vector<vector<int>>& grid){
        queue<pair<int,int>> nq;
        nq.push({i,j});
        grid[i][j]=-1;
        while(!nq.empty()){
            auto [x,y]=nq.front();
            nq.pop();
            int cnt=0;
            for(int i=0;i<4;i++){
                int xx=x+dx[i],yy=y+dy[i];
                if(xx<0||xx>=n||yy<0||yy>=n||grid[xx][yy]==0){
                    cnt++;
                    continue;
                }
                if(grid[xx][yy]==1){
                    nq.push({xx,yy});
                    grid[xx][yy]=-1;
                }
            }
            if(cnt) q.push({x,y});
        }
    }
    int shortestBridge(vector<vector<int>>& grid) {
        n=grid.size(),m=grid[0].size();
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(grid[i][j]){
                    bfs(i,j,grid);
                    break;
                }
            }
            if(!q.empty()) break;
        }
        int ans=0;
        while(!q.empty()){
            int sz=q.size();
            while(sz--){
                auto [x,y]=q.front();
                q.pop();
                for(int i=0;i<4;i++){
                    int xx=x+dx[i],yy=y+dy[i];
                    if(xx<0||xx>=n||yy<0||yy>=n||grid[xx][yy]==-1) continue;
                    else if(grid[xx][yy]==0){
                        q.push({xx,yy});
                        grid[xx][yy]=-1;
                    }
                    else return ans;
                }
            }
            ans++;
        }
        return ans;
    }
};

时间复杂度:O(n*m)。

空间复杂度:O(n*m),n和m分别为矩阵的长和宽。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值