你有一個(gè)用于表示一片土地的整數(shù)矩陣land，該矩陣中每個(gè)點(diǎn)的值代表對(duì)應(yīng)地點(diǎn)的海拔高度。若值為0則表示水域。由垂直、水平或?qū)沁B接的水域?yàn)槌靥痢３靥恋拇笮∈侵赶噙B接的水域的個(gè)數(shù)。編寫一個(gè)方法來(lái)計(jì)算矩陣中所有池塘的大小，返回值需要從小到大排序。

示例：

輸入：
[
[0,2,1,0],
[0,1,0,1],
[1,1,0,1],
[0,1,0,1]
]
輸出： [1,2,4]

代碼

class Solution {public int[] pondSizes(int[][] land) {int[][] dir=new int[][]{{-1,0},{1,0},{0,1},{0,-1},{1,1},{1,-1},{-1,1},{-1,-1}};//方向PriorityQueue<Integer> heap=new PriorityQueue<>();for(int i=0;i<land.length;i++)for(int j=0;j<land[0].length;j++)//處理0元素作為入口{if(land[i][j]>0) continue;int res=pond(land,dir,i,j);if(res>0) heap.add(res);} int num=heap.size();int[] res=new int[num];for(int i=0;i<num;i++)res[i]=heap.poll();return res;}public int pond(int[][] land,int[][] dir,int x,int y) {if(x<0||y>=land[0].length||y<0||x>=land.length||land[x][y]!=0)//不在范圍內(nèi)或不滿足水域的條件return 0;int sum=1;land[x][y]=Integer.MAX_VALUE;//置為訪問for(int[] d:dir)//遍歷所有方向{int nextX=d[0]+x,nextY=d[1]+y;sum+=pond(land,dir,nextX,nextY);}return sum;} }

總結(jié)

以上是生活随笔為你收集整理的leetcode面试题 16.19. 水域大小（深度优先搜索）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。