Count Square Submatrices with All Ones
Given a m * n matrix of ones and zeros, return how many square submatrices have all ones.
Example 1:
Input: matrix = [ [0,1,1,1], [1,1,1,1], [0,1,1,1] ] Output: 15 Explanation: There are 10 squares of side 1. There are 4 squares of side 2. There is 1 square of side 3. Total number of squares = 10 + 4 + 1 = 15.
Example 2:
Input: matrix = [ [1,0,1], [1,1,0], [1,1,0] ] Output: 7 Explanation: There are 6 squares of side 1. There is 1 square of side 2. Total number of squares = 6 + 1 = 7.
Constraints:
1 <= arr.length <= 300
1 <= arr[0].length <= 300
0 <= arr[i][j] <= 1
Solution in C++
class Solution {
public:
int n,m,cnt=0;
int mat[302][302],dp[302][302];
int dfs(int i,int j)
{
if(i>=n||j>=m)
return 0;
if(mat[i][j]==0)
return 0;
if(dp[i][j]!=-1)
return dp[i][j];
int p=dfs(i+1,j);
int q=dfs(i,j+1);
int r=dfs(i+1,j+1);
return dp[i][j]=1+min(p,min(q,r));
}
int countSquares(vector<vector<int>>& matrix) {
int ans=0;
for(int i=0;i<matrix.size();i++)
{
for(int j=0;j<matrix[i].size();j++)
{
mat[i][j]=matrix[i][j];
}
}
memset(dp,-1,sizeof(dp));
n=matrix.size();
m=matrix[0].size();
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
if(mat[i][j]==1)
{
ans+= dfs(i,j);
}}}
return ans;
}};
Comments
Post a Comment