蓝桥杯岛屿个数dfs(已找到bug)

#include <bits/stdc++.h>
#define endl "\n"
using namespace std;
int m, n;
int dx[4] = {0,-1,0,1};
int dy[4] = {-1,0,1,0};

int dxx[8] = {0,-1,0,1,-1,-1,1,1};
int dyy[8] = {-1,0,1,0,-1,1,-1,1};

int res = 0;
void dfs_road(vector<vector<char>> &grid,int x,int y,vector<vector<bool>> &visited)
{
  visited[x][y] = true;
  for(int i = 0;i < 4;i++)
  {
    int nextx = x + dx[i];
    int nexty = y + dy[i];
    if(nextx < 0 || nexty < 0 || nextx >= grid.size() || nexty >= grid[0].size()) continue;
    if(!visited[nextx][nexty] && grid[nextx][nexty] == '1')
    {
      visited[nextx][nexty] = true;
      dfs_road(grid,nextx,nexty,visited);
    }
  }
}
//
void dfs_sea(vector<vector<char>> &grid,int x,int y,vector<vector<bool>> &visited)
{
  visited[x][y] = true;
  for(int i = 0;i < 8;i++)
  {
    int nextx = x + dxx[i];
    int nexty = y + dyy[i];
    if(nextx < 0 || nexty < 0 || nextx >= grid.size() || nexty >= grid[0].size()) continue;
    if(!visited[nextx][nexty] && grid[nextx][nexty] == '1')
    {
      visited[nexty][nexty] = true;
      dfs_road(grid,nextx,nexty,visited);
      res++;
    }
    if(!visited[nextx][nexty] && grid[nextx][nexty] == '0')
    {
      visited[nextx][nexty] = true;
      dfs_sea(grid,nextx,nexty,visited);
    }
  }
}
void solve()
{
  cin >> m >>n;
  res = 0;
  vector<vector<char>> grid(m,vector<char>(n));
  vector<vector<bool>> visited = vector<vector<bool>>(m,vector<bool>(n,false));
  for(int i = 0;i < m;i++)
  {
    for(int j = 0;j < n;j++)
    {
      cin >> grid[i][j];
    }
  }
  //从边缘遍历岛屿
  for(int i = 0;i < m;i++)
  {
    if(!visited[i][0] && grid[i][0] == '1')
    {
      dfs_road(grid,i,0,visited);
      res++;
    }
    if(!visited[i][n - 1] && grid[i][n - 1] == '1')
    {
      dfs_road(grid,i,n - 1,visited);
      res++;
    }
  }
  for(int j = 0;j < n;j++)
  {
    if(!visited[0][j] && grid[0][j] == '1') 
    {
      dfs_road(grid,0,j,visited);
      res++;
    }
    if(!visited[m - 1][j] && grid[m - 1][j] == '1')
    {
      dfs_road(grid,m - 1,j,visited);
      res++;
    }
  }
  //从边缘的海遍历 遇见没有遍历过的陆地再做dfs
  for(int i = 0;i < m;i++)
  {
    if(!visited[i][0] && grid[i][0] == '0')
    {
      dfs_sea(grid,i,0,visited);
    }
    //
    if(!visited[i][n - 1] && grid[i][n - 1] == '0')
    {
      dfs_sea(grid,i,n - 1,visited);
    }
  }
  for(int j = 0;j < n;j++)
  {
    if(!visited[0][j] && grid[0][j] == '0') 
    {
      dfs_sea(grid,0,j,visited);
    }
    //
    if(!visited[m - 1][j] && grid[m - 1][j] == '0')
    {
      dfs_sea(grid,m - 1,j,visited);
    }
  }
  cout << res << endl;
}
signed main()
{
  ios::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  int t = 1;
  cin >> t;
  while(t--)
  solve();
  return 0;
}

 

 

 

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值