leetcode面试题 16.19. 水域大小（深度优先搜索）

你有一个用于表示一片土地的整数矩阵land，该矩阵中每个点的值代表对应地点的海拔高度。若值为0则表示水域。由垂直、水平或对角连接的水域为池塘。池塘的大小是指相连接的水域的个数。编写一个方法来计算矩阵中所有池塘的大小，返回值需要从小到大排序。

示例：

输入：
[
[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)//不在范围内或不满足水域的条件
            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;

    }
}