一个二维矩阵,里面只有0和1,问怎样求指定某一个格与它数字相同的连通格子个数
记录一下 感觉效率不高
int iter(int x, int y, vector<vector<int>>& nums, vector<vector<int>>& dp){
int lenX = nums.size();
int lenY = nums[0].size();
if (x < 0 || x >= lenX || y < 0 || y >= lenY)
return 0;
if (dp[x][y] == 1 || nums[x][y] == 0)
return 0;
dp[x][y] = 1;
int left = iter(x, y - 1, nums, dp);
int right = iter(x, y + 1, nums, dp);
int up = iter(x - 1, y, nums, dp);
int down = iter(x + 1, y, nums, dp);
return 1 + left + right + up + down;
}
int numsOfOnes(int x, int y, vector<vector<int>>& nums){
vector<int> zeros(nums[0].size(), 0);
vector<vector<int>> dp(nums.size(),zeros);
return iter(x, y, nums, dp);
}