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

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

又是前缀和,不过这次不一样,是斜前缀和,值得记录一下!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());
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值