LeetCode（1277）：統計全爲 1 的正方形子矩陣 Count Square Submatrices with All Ones（Java）

2019.12.29 LeetCode 從零單刷個人筆記整理（持續更新）

github：https://github.com/ChopinXBP/LeetCode-Babel

原地動態規劃。

遍歷每一個位置ij時，將該位的數值置爲從該位起左邊的1的總個數，例如對於矩陣

  [0,1,1,1],
  [1,1,1,1],
  [0,1,1,1]

遍歷結束後效果爲：

  [0,1,2,3],
  [1,2,3,4],
  [0,1,2,3]

爲了統計正方形矩陣的個數，可以假定當前位置ij爲正方形的右下角點，其值爲可能的最大邊長maxlen = matrix[i][j]。

沿着同一列j向上遍歷，遍歷行數即爲當前正方形的邊長curlen，當遍歷過程的最小邊長minlen >= curlen時，可以構成當前正方形。

---

傳送門：統計全爲 1 的正方形子矩陣

Given a m * n matrix of ones and zeros, return how many square submatrices have all ones.

給你一個 m * n 的矩陣，矩陣中的元素不是 0 就是 1，請你統計並返回其中完全由 1 組成的 正方形 子矩陣的個數。

示例 1：
輸入：matrix =
[
  [0,1,1,1],
  [1,1,1,1],
  [0,1,1,1]
]
輸出：15
解釋： 
邊長爲 1 的正方形有 10 個。
邊長爲 2 的正方形有 4 個。
邊長爲 3 的正方形有 1 個。
正方形的總數 = 10 + 4 + 1 = 15.

示例 2：
輸入：matrix = 
[
  [1,0,1],
  [1,1,0],
  [1,1,0]
]
輸出：7
解釋：
邊長爲 1 的正方形有 6 個。 
邊長爲 2 的正方形有 1 個。
正方形的總數 = 6 + 1 = 7.

---

/**
 *
 * Given a m * n matrix of ones and zeros, return how many square submatrices have all ones.
 * 給你一個 m * n 的矩陣，矩陣中的元素不是 0 就是 1，請你統計並返回其中完全由 1 組成的 正方形 子矩陣的個數。
 *
 */

public class CountSquareSubmatricesWithAllOnes {
    public int countSquares(int[][] matrix) {
        int result = 0;
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[0].length; j++) {
                if(matrix[i][j] != 1){
                    continue;
                }
                result++;
                if(j == 0){
                    matrix[i][j] = 1;
                    continue;
                }
                matrix[i][j] = matrix[i][j - 1] + 1;
                int maxlen = matrix[i][j];
                int minlen = maxlen;
                int row = i - 1;
                int curlen = 2;
                while(row >= 0 && curlen <= maxlen){
                    minlen = Math.min(matrix[row][j], minlen);
                    if(minlen < curlen){
                        break;
                    }
                    result++;
                    row--;
                    curlen++;
                }
            }
        }
        return result;
    }
}

---

#Coding一小時，Copying一秒鐘。留個言點個讚唄，謝謝你#