BFS解决FloodFill算法:(Leetcode:200. 岛屿数量)

题目链接:200. 岛屿数量 - 力扣(LeetCode) 

本题由于没有给出开始搜索的位置,所以每一个位置都要进行一次广度优先搜索 

另外为了不修改原数组数据,需要设置一个bool类型的二维数组vis来判断某个位置是否被搜索过

class Solution {
    int dx[4]={1,-1,0,0};
    int dy[4]={0,0,1,-1};
    typedef pair<int,int> PII;
    int m=0,n=0;
    vector<vector<bool>> vis;//指示数组,指示某个位置是否被搜索过
public:
    int numIslands(vector<vector<char>>& grid) {
        int ret=0;//岛屿数量
        m=grid.size();
        n=grid[0].size();
        vis.resize(m);//初始化指示数组
        for(int i=0;i<m;++i)//初始化指示数组
        {
            vis[i].resize(n);
            for(int j=0;j<n;++j) vis[i][j]=false;
        }
        //每个位置都要进行一次广度优先搜索
        for(int i=0;i<m;++i)
        {
            for(int j=0;j<n;++j)
            {
                if(grid[i][j]=='1'&&!vis[i][j])
                {
                    ++ret;
                    //广度优先搜索
                    bfs(grid,i,j);
                }
            }
        }
        return ret;
    } 
    //广度优先搜索
    void bfs(vector<vector<char>>& grid,int i,int j)
    {
        queue<PII> q;
        q.push({i,j});
        vis[i][j]=true;
        while(q.size())
        {
            int a=q.front().first;
            int b=q.front().second;
            q.pop();
            for(int k=0;k<4;++k)
            {
                int x=a+dx[k];
                int y=b+dy[k];
                if(x>=0&&x<m&&y>=0&&y<n&&grid[x][y]=='1'&&!vis[x][y])
                {
                    q.push({x,y});
                    vis[x][y]=true;
                }
            }          
        }
    } 
};
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
from pythonds.basic import Queue class Vertex: def __init__(self,key): self.id = key self.connectedTo = {} def addNeighbor(self,nbr,weight=0): self.connectedTo[nbr] = weight def __str__(self): return str(self.id) + ' connectedTo: ' + str([x.id for x in self.connectedTo]) def getConnections(self): return self.connectedTo.keys() def getId(self): return self.id def getWeight(self,nbr): return self.connectedTo[nbr] class Graph: def __init__(self): self.vertList = {} self.numVertices = 0 def addVertex(self,key): self.numVertices = self.numVertices + 1 newVertex = Vertex(key) self.vertList[key] = newVertex return newVertex def getVertex(self,n): if n in self.vertList: return self.vertList[n] else: return None def __contains__(self,n): return n in self.vertList def addEdge(self,f,t,cost=0): if f not in self.vertList: nv = self.addVertex(f) if t not in self.vertList: nv = self.addVertex(t) self.vertList[f].addNeighbor(self.vertList[t], cost) def getVertices(self): return self.vertList.keys() def __iter__(self): return iter(self.vertList.values()) def bfs(g,start): start.setDistance(0) start.setPred(None) vertQueue=Queue() vertQueue.enqueue(start) while (vertQueue.size()>0): currentVert=vertQueue.dequeue() for nbr in currentVert.getConnections(): if (nbr.getColor()=='White'): nbr.setColor('gray') nbr.setDistance(currentVert.getDistance()+1) nbr.setPred(currentVert) vertQueue.enqueue(nbr) currentVert.setColor('black') List=["""1:A,2:B,3:C,4:D,5:E,6:F"""] g=Graph() for i in range(6): g.addVertex(i) g.addEdge(1,2,7) g.addEdge(2,1,2) g.addEdge(1,3,5) g.addEdge(1,6,1) g.addEdge(2,4,7) g.addEdge(2,5,3) g.addEdge(3,2,2) g.addEdge(3,6,8) g.addEdge(4,1,1) g.addEdge(4,5,2) g.addEdge(4,6,4) g.addEdge(5,1,6) g.addEdge(5,4,5) g.addEdge(6,2,1) g.addEdge(6,5,8) bfs(g,)优化这段代码
06-12

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

南林yan

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值