leetcode 水域大小（C++）

你有一个用于表示一片土地的整数矩阵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

C++

class Solution {
public:
    void dfs(vector<vector<int>>& land, int i, int j) {
        for(int k=0;k<8;k++) {
            int y=i+dy[k];
            int x=j+dx[k];
            if(y>=0 && y<land.size() && x>=0 && x<land[0].size() && 0==land[y][x]) {
                count++;
                land[y][x]=3;
                dfs(land,y,x);
            }
        }
    }
    
    vector<int> pondSizes(vector<vector<int>>& land) {
        int m=land.size();
        int n=land[0].size();
        vector<int> res;
        for(int i=0;i<m;i++) {
            for(int j=0;j<n;j++) {
                if(0==land[i][j]) {
                    count=1;
                    land[i][j]=3;
                    dfs(land,i,j);
                    res.push_back(count);
                }
            }
        }
        sort(res.begin(),res.end());
        return res;
    }
private:
    int dy[8]={-1,-1,-1,0,0,1,1,1};
    int dx[8]={-1,0,1,-1,1,-1,0,1};
    int count=0;
};