LeetCode 695. 島嶼的最大面積【dfs】

題目描述

給定一個包含了一些 0 和 1 的非空二維數組 grid 。

一個 島嶼 是由一些相鄰的 1 (代表土地) 構成的組合，這裏的「相鄰」要求兩個 1 必須在水平或者豎直方向上相鄰。你可以假設 grid 的四個邊緣都被 0（代表水）包圍着。

找到給定的二維數組中最大的島嶼面積。(如果沒有島嶼，則返回面積爲 0 。)

示例 1:

[[0,0,1,0,0,0,0,1,0,0,0,0,0],
[0,0,0,0,0,0,0,1,1,1,0,0,0],
[0,1,1,0,1,0,0,0,0,0,0,0,0],
[0,1,0,0,1,1,0,0,1,0,1,0,0],
[0,1,0,0,1,1,0,0,1,1,1,0,0],
[0,0,0,0,0,0,0,0,0,0,1,0,0],
[0,0,0,0,0,0,0,1,1,1,0,0,0],
[0,0,0,0,0,0,0,1,1,0,0,0,0]]

對於上面這個給定矩陣應返回 6。注意答案不應該是 11 ，因爲島嶼只能包含水平或垂直的四個方向的 1 。

示例 2:

[[0,0,0,0,0,0,0,0]]

對於上面這個給定的矩陣, 返回 0。

注意: 給定的矩陣grid 的長度和寬度都不超過 50。

解題思路

十分標準的搜索題，用來練手深度優先搜索。

AC

class Solution {
public:
    vector<int> dir{-1,0,1,0,-1};
    int maxAreaOfIsland(vector<vector<int>>& grid) {
        if(grid.empty() || grid[0].empty()) return 0;
        int max_area=0;
        for(int i=0;i<grid.size();i++){
            for(int j=0;j<grid[0].size();j++){
                if(grid[i][j]==1){
                    max_area=max(max_area,dfs(grid,i,j));
                }
            }
        }
        return max_area;
    }
    int dfs(vector<vector<int>>& grid,int x,int y){
        if(grid[x][y]==0) return 0;
        grid[x][y]=0;
        int xx,yy,ans=1;
        for(int i=0;i<4;i++){
            xx=x+dir[i],yy=y+dir[i+1];
            if(xx>=0&&xx<grid.size() && yy>=0&&yy<grid[0].size()){
                ans+=dfs(grid,xx,yy);
            }
        }
        return ans;
    }
};