给你一个大小为 m x n 的整数矩阵 grid 。
按以下形式将矩阵的一部分定义为一个 沙漏 :
返回沙漏中元素的 最大 总和。
注意:沙漏无法旋转且必须整个包含在矩阵中。
示例 1:
输入:grid = [[6,2,1,3],[4,2,1,5],[9,2,8,7],[4,1,2,9]]
输出:30
解释:上图中的单元格表示元素总和最大的沙漏:6 + 2 + 1 + 2 + 9 + 2 + 8 = 30 。
示例 2:
输入:grid = [[1,2,3],[4,5,6],[7,8,9]]
输出:35
解释:上图中的单元格表示元素总和最大的沙漏:1 + 2 + 3 + 5 + 7 + 8 + 9 = 35 。
提示:
m == grid.length
n == grid[i].length
3 <= m, n <= 150
0 <= grid[i][j] <= 106
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-sum-of-an-hourglass
思路:
枚举每一个沙漏,但是我们可以用前缀和计算一个3*3正方形然后剪掉那两块。
AC代码:
class Solution {
public:
#define N 155
typedef long long LL;
LL a[N][N],s[N][N];
int maxSum(vector<vector<int>>& g) {
int n = g.size();
int m = g[0].size();
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
a[i][j]=g[i-1][j-1];
s[i][j]=a[i][j]+s[i-1][j]+s[i][j-1]-s[i-1][j-1];
}
LL res=0;
for(int i=3;i<=n;i++)
for(int j=3;j<=m;j++){
int x1=i-2;
int y1=j-2;
LL sum = s[i][j]-s[x1-1][j]-s[i][y1-1]+s[x1-1][y1-1];
sum=sum-a[i-1][j]-a[i-1][j-2];
res = max(sum,res);
}
return res;
}
};