又是前缀和,不过这次不一样,是斜前缀和,值得记录一下!btw,y总太强了,讲得太清晰了!
LC5757:矩阵中最大的三个菱形和
const int N = 110;
class Solution {
public:
int s1[N][N];
int s2[N][N];
vector<int> getBiggestThree(vector<vector<int>>& g) {
int m = g.size();
int n = g[0].size();
for(int i = 1; i <= m; i++)
{
for(int j = 1; j <= n; j++)
{
s1[i][j] = s1[i - 1][j - 1] + g[i - 1][j - 1];
s2[i][j] = s2[i - 1][j + 1] + g[i - 1][j - 1];
}
}
set<int> S;
for(int i = 1; i <= m; i++)
{
for(int j = 1; j <= n; j++)
{
S.insert(g[i - 1][j - 1]);
for(int k = 1; i + k <= m && i - k >= 1 && j + k <= n && j - k >= 1; k++)
{
int a = s2[i][j - k] - s2[i - k][j];
int b = s1[i][j + k] - s1[i - k][j];
int c = s1[i + k][j] - s1[i][j - k];
int d = s2[i + k][j] - s2[i][j + k];
S.insert(a + b + c + d + g[i - k - 1][j - 1] - g[i + k - 1][j - 1]);
}
}
}
while(S.size() > 3) S.erase(S.begin());
return vector<int>(S.rbegin(), S.rend());
}
};