Leetcode 363. 矩形区域不超过 K 的最大数值和 (二维前缀和以及set优化）

最新推荐文章于 2023-05-24 11:22:14 发布

wwxy261

最新推荐文章于 2023-05-24 11:22:14 发布

阅读量138

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/116016171

版权

二维前缀和可以在O(nm)的时间复杂度预处理下，在O(1)时间求出任意区域的元素之和。

所以枚举四个边界，在O(1)时间求出区域和，和目标值比较，求最大值，这样的时间复杂度是O(n^2m^2)，超时

class Solution {
public:
    vector<vector<int>> sums;
    int sumRegion(int row1, int col1, int row2, int col2) {
        return sums[row2][col2] - sums[row1-1][col2] - sums[row2][col1 - 1] + sums[row1 - 1][col1 - 1];
    }

    int maxSumSubmatrix(vector<vector<int>>& matrix, int K) {
        // 枚举下边界和左右边界，对左右边界进行求和处理
        // 找出最合适的上边界
        int n = matrix.size(), m = matrix[0].size();
        sums.resize(n+1, vector<int>(m+1));
        for(int i = 1; i <= n; i++){
            for(int j = 1; j <= m; j++){
                sums[i][j] = sums[i - 1][j]