算法练习--水域大小__深度优先

题目:

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

示例:

输入:
[
  [0,2,1,0],
  [0,1,0,1],
  [1,1,0,1],
  [0,1,0,1]
]
输出: [1,2,4]

个人思路:

深度优先,遍历数组,如果为0,那么对其进行深度优先的递归,递归它上下左右和对角线,然后返回面积,再和当前最大面积进行比较,更新最大面积。

代码:

class Solution {
    public int[] pondSizes(int[][] land) {
        ArrayList list=new ArrayList();
        for(int i=0;iland.length-1||j<0||j>land[0].length-1||land[i][j]!=0){
            return 0;
        }else{
            land[i][j]=-1;
            return getSize(land,i-1,j)+getSize(land,i+1,j)+getSize(land,i,j+1)+getSize(land,i,j-1)+getSize(land,i+1,j-1)+getSize(land,i+1,j+1)+getSize(land,i-1,j-1)+getSize(land,i-1,j+1)+1;
        }
    }
}

 

你可能感兴趣的:(java算法练习)