算法練習--水域大小__深度優先

題目:

你有一個用於表示一片土地的整數矩陣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<Integer> list=new ArrayList();
        for(int i=0;i<land.length;i++){
            for(int j=0;j<land[0].length;j++){
                if(land[i][j]==0){
                    int size=getSize(land,i,j);
                    list.add(size);
                }
            }
        }
        int[] re=new int[list.size()];
        for(int i=0;i<list.size();i++){
            re[i]=list.get(i);
        }
        Arrays.sort(re);
        return re;
    }
    public int getSize(int[][] land,int i,int j){
        if(i<0||i>land.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;
        }
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章