leetcode 4405.统计子矩阵

这道题其实就是运用二维前缀和进行计算的。

如果说要进行暴力,那就是四重循环来凑活:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cstdlib>
#include<cmath> 
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<sstream>
#include<map>
#include<limits.h>
#include<set>
#define MAX 505
#define _for(i,a,b) for(int i=a;i<(b);i++)
#define ALL(x) x.begin(),x.end()
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
int arr[MAX][MAX];
int b[MAX][MAX];
LL n, m, counts, num;
void qianzhuihe() {
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            b[i][j] = b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1] + arr[i][j];
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m >> num;
    _for(i, 1, n+1) {
        _for (j, 1, m + 1) {
            cin >> arr[i][j];
        }
    }
    qianzhuihe();
    for (int x1 = 1; x1 <= n; x1++) {
        for (int y1 = 1; y1 <= m; y1++) {//定下一个固定点
            for (int x2 = x1; x2 <= n; x2++) {
                for (int y2 = y1; y2 <= m; y2++) {
                    if (b[x2][y2] - b[x1 - 1][y2] - b[x2][y1 - 1] + b[x1-1][y1-1] <= num)
                        counts++;
                }//依次遍历其他的子矩阵
            }
        }
    }
    cout << counts;
    return 0;
}

所以我们需要进行优化才行:

这里我们的优化用到了双指针的思路。

那么,到底为什么需要这样用呢?我接下来就讲一讲具体的思路:

首先,我们知道,在对于矩阵进行遍历的时候,我们是有四个边界的,也就是上界下界,左界右界一共就这4个边界,要想让时间复杂度降低的话,我们思考怎么才能减少一层循环才行。

假设,我们把上下两个边界进行了固定,那么剩下的左右边界移动,这个时候是不是就是在这一个小边界里进行了遍历呢?也就是说,我们把这个问题转变成了动态的一维前缀和问题了。

也就是说,如果上下边界固定住了,假如上边界是1,下边界是2,这个时候这两个边界之间是不是就只剩下了一行数据了呢?这样是不是一维的前缀和考虑了呢?显然是的。但是,如果说上下边界分别是4,1的话,中间不就是有3行了吗?这样该怎么办呢?我们类比着,将着每一列的数加起来,这样不就变成了一列一个数了吗?这样不就变相的转变成了一维数组的问题了么?

其实这样就已经可以了,我们这个时候定义两个指针,进行一前一后的遍历,显然要比我们一直遍历要高效,也就是左指针和右指针。双指针能够根据判断当前的权值是否能够达到题目中的条件做出判断,如果说此时的权值要大,说明我们需要缩小现在的区间,让左指针向着右指针的方向靠拢。

有人问,为啥不是右指针开始往后找呢?这里我们的右指针其实也是从跟左指针一样的初始位置,你如果这样认为,那有可能是其他的双指针题目让你把右指针放在了最后面,其实这样也行,不妨尝试一下。我们就只按照现在这个思路说。

上代码:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cstdlib>
#include<cmath> 
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<sstream>
#include<map>
#include<limits.h>
#include<set>
#define MAX 505
#define _for(i,a,b) for(int i=a;i<(b);i++)
#define ALL(x) x.begin(),x.end()
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
int arr[MAX][MAX];
int b[MAX][MAX];
LL n, m, counts, num;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m >> num;
    _for(i, 1, n+1) {
        _for (j, 1, m + 1) {
            cin >> arr[i][j];
            arr[i][j] += arr[i - 1][j];
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = i; j <= n; j++) {
            for (int l = 1, r = 1, sum = 0; r <= m; r++) {
                sum += arr[j][r] - arr[i - 1][r];
                while (sum > num) {
                    sum -= arr[j][l] - arr[i - 1][l];
                    l++;
                }
                counts += r - l + 1;
            }
        }
    }
    cout << counts << endl;
    return 0;
}
相关推荐
积兆科技19 分钟前
从汽车企业案例看仓网规划的关键步骤(视频版)
人工智能·算法·汽车·制造
Lenyiin1 小时前
01.01、判定字符是否唯一
算法·哈希算法
闻缺陷则喜何志丹1 小时前
【C++动态规划】1458. 两个子序列的最大点积|1823
c++·算法·动态规划·力扣·最大·子序列·点积
三万棵雪松1 小时前
3.系统学习-熵与决策树
学习·算法·决策树
半盏茶香1 小时前
C语言勘破之路-最终篇 —— 预处理(上)
c语言·开发语言·数据结构·c++·算法
2401_858286111 小时前
118.【C语言】数据结构之排序(堆排序和冒泡排序)
c语言·数据结构·算法
不会kao代码的小王2 小时前
从零开始搭建 AI 音乐生成器:MusicGPT 的超简单部署指南
科技·算法·开源·powerpoint
就爱学编程2 小时前
重生之我在异世界学编程之数据结构与算法:单链表篇
数据结构·算法·链表
云云3216 小时前
云手机服务器如何做到群控多台手机的?
服务器·线性代数·安全·智能手机·矩阵
88号技师9 小时前
2024年12月一区SCI-加权平均优化算法Weighted average algorithm-附Matlab免费代码
人工智能·算法·matlab·优化算法