LeetCode 面試題 16.19. 水域大小 （DFS）

題目鏈接：https://leetcode-cn.com/problems/pond-sizes-lcci/

你有一個用於表示一片土地的整數矩陣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

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int t;
int cmp(void* a,void* b)
{
    return *(int*)a-*(int*)b;
}
void dfs(int** land, int landSize, int* landColSize,int x,int y)
{
    int i,j;
    t++;
    for(int dx=-1;dx<=1;dx++){
        for(int dy=-1;dy<=1;dy++){
            int nx=x+dx,ny=y+dy;
            if(nx>=0&&nx<landSize&&ny>=0&&ny<landColSize[0]&&land[nx][ny]==0){
                land[nx][ny]=1;
                dfs(land,landSize,landColSize,nx,ny);
            }
        }
    }
}
int* pondSizes(int** land, int landSize, int* landColSize, int* returnSize){
    if(land==NULL||landSize==0){
        *returnSize=0;
        return NULL;
    }
    int* area=(int*)malloc(sizeof(int)*(landSize*landColSize[0]));
    int cnt=0,i,j;
    for(i=0;i<landSize;i++){
        for(j=0;j<landColSize[0];j++){
            if(land[i][j]==0){
                t=0;
                dfs(land,landSize,landColSize,i,j);
                area[cnt++]=t-1;
            }
        }
    }
    *returnSize=cnt;
    qsort(area,cnt,sizeof(int),cmp);
    return area;
}