leetcode1504. 统计全 1 子矩形(动态规划)
生活随笔
收集整理的這篇文章主要介紹了
leetcode1504. 统计全 1 子矩形(动态规划)
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
給你一個只包含 0 和 1 的 rows * columns 矩陣 mat ,請你返回有多少個 子矩形 的元素全部都是 1 。
示例 1:
輸入:mat =
[[1,0,1],
[1,1,0],
[1,1,0]]
輸出:13
解釋:
有 6 個 1x1 的矩形。
有 2 個 1x2 的矩形。
有 3 個 2x1 的矩形。
有 1 個 2x2 的矩形。
有 1 個 3x1 的矩形。
矩形數(shù)目總共 = 6 + 2 + 3 + 1 + 1 = 13 。
解題思路
數(shù)組含義:dp[i][j]位于(i,j)的元素向左延長的長度
狀態(tài)轉(zhuǎn)移:min= Math.min(dp[k][j],min) 向上遍歷,加入滿足最小長度的矩形
代碼
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 子矩形(动态规划)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: leetcode剑指 Offer 14-
- 下一篇: leetcode264. 丑数 II