#LC5757：矩阵中最大的三个菱形和

最新推荐文章于 2025-07-28 10:30:49 发布

原创最新推荐文章于 2025-07-28 10:30:49 发布 · 264 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode & AcWing 专栏收录该内容

37 篇文章

订阅专栏

博客介绍了如何利用前缀和与斜前缀和解决LeetCode上的5757题，即找到矩阵中最大的三个菱形和。通过构建s1和s2二维数组来存储斜前缀和，然后插入集合S进行求解。这种方法展示了高效处理矩阵问题的思路，适合对数据结构和算法感兴趣的读者学习。

又是前缀和，不过这次不一样，是斜前缀和，值得记录一下！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());
    }
};