給你一個(gè)只包含 0 和 1 的 rows * columns 矩陣 mat ，請(qǐng)你返回有多少個(gè) 子矩形 的元素全部都是 1 。

示例 1：
輸入：mat =
[[1,0,1],
[1,1,0],
[1,1,0]]
輸出：13
解釋：
有 6 個(gè) 1x1 的矩形。
有 2 個(gè) 1x2 的矩形。
有 3 個(gè) 2x1 的矩形。
有 1 個(gè) 2x2 的矩形。
有 1 個(gè) 3x1 的矩形。
矩形數(shù)目總共 = 6 + 2 + 3 + 1 + 1 = 13 。

解題思路

數(shù)組含義：dp[i][j]位于（i，j）的元素向左延長(zhǎng)的長(zhǎng)度
狀態(tài)轉(zhuǎn)移：min= Math.min(dp[k][j],min) 向上遍歷，加入滿足最小長(zhǎng)度的矩形

代碼

class Solution {public int numSubmat(int[][] mat) {int n=mat.length,m=mat[0].length,res=0;int[][]dp=new int[n][m+1];for(int i=0;i<n;i++)for(int j=1;j<=m;j++)dp[i][j]=mat[i][j-1]==1?dp[i][j-1]+1:0;for(int i=0;i<n;i++)for(int j=1;j<=m;j++){int min=Integer.MAX_VALUE;for (int k=i;k>=0;k--)if(dp[k][j]==0) break;else{min= Math.min(dp[k][j],min);res+=min;}}return res;} }

總結(jié)

以上是生活随笔為你收集整理的leetcode1504. 统计全 1 子矩形（动态规划）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。