给你一个 m x n
的整数矩阵 grid
。
菱形和 指的是 grid
中一个正菱形 边界 上的元素之和。本题中的菱形必须为正方形旋转45度,且四个角都在一个格子当中。下图是四个可行的菱形,每个菱形和应该包含的格子都用了相应颜色标注在图中。
注意,菱形可以是一个面积为 0 的区域,如上图中右下角的紫色菱形所示。
请你按照 降序 返回 grid
中三个最大的 互不相同的菱形和 。如果不同的和少于三个,则将它们全部返回。
示例 1:
输入:grid = [[3,4,5,1,3],[3,3,4,2,3],[20,30,200,40,10],[1,5,5,4,1],[4,3,2,2,5]]
输出:[228,216,211]
解释:最大的三个菱形和如上图所示。
- 蓝色:20 + 3 + 200 + 5 = 228
- 红色:200 + 2 + 10 + 4 = 216
- 绿色:5 + 200 + 4 + 2 = 211
示例 2:
输入:grid = [[1,2,3],[4,5,6],[7,8,9]]
输出:[20,9,8]
解释:最大的三个菱形和如上图所示。
- 蓝色:4 + 2 + 6 + 8 = 20
- 红色:9 (右下角红色的面积为 0 的菱形)
- 绿色:8 (下方中央面积为 0 的菱形)
示例 3:
输入:grid = [[7,7,7]]
输出:[7]
解释:所有三个可能的菱形和都相同,所以返回 [7] 。
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 100
1 <= grid[i][j] <= 105
C++
class Solution {
public:
int compute(vector<vector<int>>& grid, int r, int c, int k) {
int sum=0;
int left=grid[r][c-k];
int right=grid[r][c+k];
int top=grid[r-k][c];
int bottom=grid[r+k][c];
for(int i=r-1,j=c-k+1;i>r-k,j-c;i--,j++) {
sum+=grid[i][j];
}
for(int i=r-k+1,j=c+1;i<r,j<c+k;i++,j++) {
sum+=grid[i][j];
}
for(int i=r+1,j=c+k-1;i<r+k,j>c;i++,j--) {
sum+=grid[i][j];
}
for(int i=r+1,j=c-k+1;i<r+k,j<c;i++,j++) {
sum+=grid[i][j];
}
return sum+left+right+top+bottom;
}
vector<int> getBiggestThree(vector<vector<int>>& grid) {
vector<int> res;
int m=grid.size();
int n=grid[0].size();
int k=(min(m,n)-1)/2;
for(int i=0;i<m;i++) {
for(int j=0;j<n;j++) {
res.push_back(grid[i][j]);
for(int s=1;s<=k;s++) {
if(i-s>=0 && i+s<m && j-s>=0 && j+s<n) {
int sum=compute(grid,i,j,s);
res.push_back(sum);
}
}
}
}
unordered_set<int> st;
for(auto it:res) {
st.insert(it);
}
vector<int> ans;
for(auto it:st) {
ans.push_back(it);
}
if(ans.size()<=3) {
return ans;
}
int len=ans.size();
sort(ans.begin(),ans.end());
return {ans[len-1],ans[len-2],ans[len-3]};
}
};