leetcode 面试题 16.19. 水域大小

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

示例:

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

0 < len(land) <= 1000
0 < len(land[i]) <= 1000
通过次数9,673提交次数15,987


class Solution {
    int n,m;
public:
    vector<int> pondSizes(vector<vector<int>>& land) {
          n=land.size();
          m=land[0].size();
        
        vector<int> vi;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(land[i][j]==0)
                {
                    //cout<
                    int k=0;
                    dfs(i,j,k,land);
                    if(k!=0)
                      vi.push_back(k);
                }
            }
        }
        sort(vi.begin(),vi.end());
        return vi;


    }

    void dfs(int x,int y,int &k,vector<vector<int>> &land)
    {
        land[x][y]=-1;
        k++;
        int dx[8]={-1,-1,0,1,1,1,0,-1};
        int dy[8]={0,1,1,1,0,-1,-1,-1};//顺序是从正上顺时针旋转
        
        for(int i=0;i<8;i++)
        {
            int xx=dx[i]+x;
            int yy=dy[i]+y;
            if(xx>=0&&xx<n && yy>=0&&yy<m && land[xx][yy]==0)//这里yy>=0,没写=,要细心啊
               dfs(xx,yy,k,land);

        }


    }
};

你可能感兴趣的:(近期试题,leetcode,算法)