原题链接:https://leetcode-cn.com/problems/largest-1-bordered-square/
相关题目:最大加号标志
动态规划
dpx[i][j]表示x方向从i到j边长长度
dpy[i][j]表示y方向从i到j边长长度
计算正方形的元素个数,就是计算正方形的最大边长,对于任何一个点(i,j),我们判断以它为左上角的正方形,四条边是否都为1即可。我们不用依次遍历,因为边长相同,所以我们维护一个水平和一个竖直的累加数组,用累加和相减判断是否等于边长即可。
代码:
int largest1BorderedSquare(vector<vector<int>>& grid) {
int r = grid.size();
int c = grid[0].size();
vector<vector<int>> dpx(r + 1, vector<int>(c + 1, 0));
vector<vector<int>> dpy(r + 1, vector<int>(c + 1, 0));
for (int i = 1; i <= r; i++){
for (int j = 1; j <= c; j++){
dpx[i][j] = dpx[i][j - 1] + grid[i - 1][j - 1];//水平方向的累加和
dpy[i][j] = dpy[i - 1][j] + grid[i - 1][j - 1];//垂直方向的累加和
}
}
int ans = 0;
for(int len=min(r,c);len>=1;len--){//正方形边长
if(len<=ans) break;
for (int i = 1; i <= r-len+1; i++){//(i,j)为正方形左上角坐标
for (int j = 1; j <= c-len+1; j++){
if (dpx[i][j+len-1] - dpx[i][j-1] != len){//判断上边
continue;
}
if (dpx[i + len-1][j+len-1] - dpx[i + len-1][j-1] != len){//判断下边
continue;
}
if (dpy[i + len-1][j] - dpy[i-1][j] != len){//判断左边
continue;
}
if (dpy[i + len-1][j+len-1] - dpy[i-1][j+len-1] != len){//判断右边
continue;
}
ans = len;
}
}
}
return ans*ans;
}