1. 題目

你有一個用于表示一片土地的整數矩陣 land，該矩陣中每個點的值代表對應地點的海拔高度。
若值為0則表示水域。由垂直、水平或對角連接的水域為池塘。
池塘的大小是指相連接的水域的個數。
編寫一個方法來計算矩陣中所有池塘的大小，返回值需要從小到大排序。

示例： 輸入： [[0,2,1,0],[0,1,0,1],[1,1,0,1],[0,1,0,1] ] 輸出： [1,2,4]提示： 0 < len(land) <= 1000 0 < len(land[i]) <= 1000

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/pond-sizes-lcci
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

2.1 BFS

• 該題有8個可走方向，訪問標志可以直接改地圖為 1

class Solution {int m, n;int k, x, y, x0, y0;vector<int> water;queue<vector<int>> q;vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0},{1,1},{-1,-1},{1,-1},{-1,1}}; public:vector<int> pondSizes(vector<vector<int>>& land) {m = land.size(), n = land[0].size();int i, j;for(i = 0; i < m; i++){for(j = 0; j < n; j++){if(land[i][j]==0){bfs(land, i, j, 0);}}}sort(water.begin(), water.end());return water;}void bfs(vector<vector<int>>& land, int i, int j, int count){q.push({i,j});land[i][j] = 1;//訪問過了count++;while(!q.empty()){x0 = q.front()[0];y0 = q.front()[1];q.pop();for(k = 0; k < 8; ++k){x = x0 + dir[k][0];y = y0 + dir[k][1];if(x>=0 && x<m && y>=0 && y<n && land[x][y]==0){count++;q.push({x,y});land[x][y] = 1;//訪問過了}}}water.push_back(count);} };

2.2 DFS

class Solution {int m, n;vector<int> water;queue<vector<int>> q;vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0},{1,1},{-1,-1},{1,-1},{-1,1}}; public:vector<int> pondSizes(vector<vector<int>>& land) {m = land.size(), n = land[0].size();int i, j, count;for(i = 0; i < m; i++){for(j = 0; j < n; j++){if(land[i][j]==0){count = 1;land[i][j] = 1;//訪問過了dfs(land, i, j, count);water.push_back(count);}}}sort(water.begin(), water.end());return water;}void dfs(vector<vector<int>>& land, int i, int j, int& count){int k, x, y;for(k = 0; k < 8; ++k){x = i + dir[k][0];y = j + dir[k][1];if(x>=0 && x<m && y>=0 && y<n && land[x][y]==0){count++;land[x][y] = 1;//訪問過了dfs(land,x,y,count);}}} };

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 16.19. 水域大小（BFS/DFS）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。