面试题 16.19. 水域大小

面试题 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

  每遇到一个0就从当前的0开始进行深度优先搜索，每在周围遇到一个0就将水域大小加1，当将一片连通的水域计算完后，将水域大小放入结果中，对其余部分继续计算。最后将所得结果从小到大排序一下返回即可。

class Solution {
public:
	vector<int> pondSizes(vector<vector<int>>& land) {
		vector<int> res;
		for (int i = 0; i < land.size(); i++) {
			for (int j = 0; j < land[0].size(); j++) {
				if (land[i][j] == 0) {
					int num = 0;
					spare(land, i, j, num);
					res.push_back(num);
				}
			}
		}
		sort(res.begin(), res.end());
		return res;
	}

	void spare(vector<vector<int>>& v, int j, int k, int& num) {
		if (j > v.size()-1 || k > v[0].size()-1) return;
		if (v[j][k] != 0) return;
		else {
			num++;
			v[j][k] = -1;
			spare(v, j + 1, k, num);
			spare(v, j + 1, k + 1, num);
			spare(v, j + 1, k - 1, num);
			spare(v, j, k - 1, num);
			spare(v, j, k + 1, num);
			spare(v, j - 1, k, num);
			spare(v, j - 1, k + 1, num);
			spare(v, j - 1, k - 1, num);
		}

	}
};