使用 DFS 寻找岛屿数 (C++)


在本题中,我们给定了一个 2D 二进制矩阵。我们的任务是使用 DFS 找到岛屿数量。

岛屿是指矩阵中 1 或多个相连的 1。

让我们举个例子来理解该问题:

Input : bin[][] = {{ 1 0 0 0}
   {0 1 0 1}
   {0 0 0 0}
   {0 0 1 0}}
Output : 3

Explanation

Islands are −bin00 - bin11

bin13

bin32

解题思路

要使用 DFS 来解决问题,我们将使用 DFS 技术探索矩阵中所有的相邻位置(每个数字可能多达 8 个相邻位置),并查找 1。如果我们遇到一个未访问过的值为 1,那么我们将考虑它。我们将对已访问的值进行检查,以避免重复访问。通过这样做,我们可以计算矩阵中存在的岛屿数量。

了解图上的 DFS

了解图

示例

#include <bits/stdc++.h>
using namespace std;
#define ROW 5
#define COL 5
int canVisit(int bin[][COL], int row, int col, bool visited[][COL]){
   return (row >= 0) && (row < ROW) && (col >= 0) && (col < COL) && (bin[row][col] && !visited[row][col]);
}
void DFS(int bin[][COL], int row, int col, bool visited[][COL]){
   static int getNeighbourRow[] = { -1, -1, -1, 0, 0, 1, 1, 1 };
   static int getNeighbourCol[] = { -1, 0, 1, -1, 1, -1, 0, 1 };
   visited[row][col] = true;
   for (int k = 0; k < 8; ++k)
      if (canVisit(bin, row + getNeighbourRow[k], col + getNeighbourCol[k], visited))
         DFS(bin, row + getNeighbourRow[k], col + getNeighbourCol[k], visited);
}
int findIslandCount(int bin[][COL]){
   bool visited[ROW][COL];
   memset(visited, 0, sizeof(visited));
   int islandCount = 0;
   for (int i = 0; i < ROW; ++i)
   for (int j = 0; j < COL; ++j)
   if (bin[i][j] && !visited[i][j]) {
      DFS(bin, i, j, visited);
      islandCount++;
   }
   return islandCount;
}
int main(){
   int bin[][COL] = {{1, 0, 0, 0},
   {0, 1, 0, 1},
   {0, 0, 0, 0},
   {0, 0, 1, 0}};
   cout<<"The number of islands present in the matrix is "<<findIslandCount(bin);
   return 0;
}

输出

The number of islands present in the matrix is 4

更新时间:11-Feb-2022

215 次浏览

开启您的职业

完成课程即可获得认证

开始吧
广告
© . All rights reserved.