1. 程式人生 > 實用技巧 >LeetCode 1139. 最大的以 1 為邊界的正方形 字首和

LeetCode 1139. 最大的以 1 為邊界的正方形 字首和

地址https://leetcode-cn.com/problems/largest-1-bordered-square/

給你一個由若干 01 組成的二維網格grid,請你找出邊界全部由 1 組成的最大 正方形 子網格,並返回該子網格中的元素數量。如果不存在,則返回 0。



示例 1:

輸入:grid = [[1,1,1],[1,0,1],[1,1,1]]
輸出:9
示例 2:

輸入:grid = [[1,1,0,0]]
輸出:1


提示:

1 <= grid.length <= 100
1 <= grid[0].length <= 100
grid[i][j] 為01

演算法1

使用字首和記錄 行與列到達該座標1的總數 加快查詢效率
然後遍歷每個正方形 檢視每條邊的1的總數是否達到長度判斷邊是否全1
示意圖

class Solution {
public:


int presumCol[150][150];
int presumRow[150][150];
int ans =0;
int n, m;

bool CheckSquard(int startx, int starty, int len)
{
    int endx = startx + len ;
    int endy = starty + len ;

    startx++; starty++;

    
if (endx <= n && endy <= m) { if (presumRow[startx][endy] - presumRow[startx][starty-1] == len && presumRow[endx][endy] - presumRow[endx][starty - 1] == len && presumCol[endx][starty] - presumCol[startx-1][starty] == len && presumCol[endx][endy]
- presumCol[startx-1][endy] == len) { ans = len * len; return true; } } return false; } int largest1BorderedSquare(vector<vector<int>>& grid) { if (grid.empty() || grid[0].empty()) return 0; n = grid.size(); m = grid[0].size(); //計算行列為單位的 前面有多少個1 的字首和 for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { presumRow[i][j] = presumRow[i][j - 1] + grid[i - 1][j - 1]; } } for (int y = 1; y <= m; y++) { for (int x = 1; x <= n; x++) { presumCol[x][y] = presumCol[x-1][y] + grid[x - 1][y - 1]; } } int lenlimit = min(n, m); for (int len = lenlimit; len > 0; len--) { for (int i = 0; i < n; i++) { for (int j = 0; j < m; j++) { if (CheckSquard(i, j, len)) { return ans; } } } } return ans; } };