【acwing】统计子矩阵

4405.统计子矩阵

给定一个 N×M 的矩阵 A，请你统计有多少个子矩阵 (最小 1×1，最大 N×M) 满足子矩阵中所有数的和不超过给定的整数 K?
输入格式
第一行包含三个整数 N,M 和 K。
之后 N 行每行包含 M 个整数，代表矩阵 A。
输出格式
一个整数代表答案。
数据范围
对于 30% 的数据，N,M≤20，
对于 70% 的数据，N,M≤100，
对于 100% 的数据，1≤N,M≤500;0≤Aij≤1000;1≤K≤2.5×108。
输入样例：
3 4 10
1 2 3 4
5 6 7 8
9 10 11 12
输出样例：
19
样例解释
满足条件的子矩阵一共有 19，包含：
大小为 1×1 的有 10 个。
大小为 1×2 的有 3 个。
大小为 1×3 的有 2 个。
大小为 1×4 的有 1 个。
大小为 2×1 的有 3 个。

c++一般有一个时间限制，限制为1s时，计算量在1个亿以内就不会超时
按照暴力做法的话需要四维，时间复杂度为O（n⁴），我们可以先枚举上下边界，然后就会变成一个一维问题，然后求有多少个区间的总和不超过k

由题意可知矩阵A中所有数均大于等于0，故当右边界确定后（即图中R），L指针越往左，矩阵的和越大，即单调，那么我们就可以用双指针或二分，如果用二分的话，时间复杂度为o（n^3logn），感觉也会超时，所以我们先来试一下双指针，如果可以用双指针的话，那就表示可以降到O(n³)
双指针的话要看一下两个指针是不是单调的，即R往右走的时候，L是不是单调往右走，那么我们的L需要用R来定义，所以L一般是R的一个函数，L=f（R),定义完这个函数之后，我们再来证明这个函数是单调的，如果单调的话，我们才可以使用双指针
注：双指针的单调和二分的单调不同，双指针的单调是一个指针随另外一个指针单调，二分的单调指数值随某一个数单调

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 510;
int s[N][N]; //尽管是二维数组，实际上表示一维前缀和
int m,n,k;
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    // 求解一维前缀和。s[][j]表示第j列的各元素之和
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            scanf("%d",&s[i][j]);
            s[i][j]+=s[i-1][j]; 
        }
    long long res=0;
    for (int i=1;i<=n;i++)  //枚举第一个横坐标x1范围
        for (int j=i;j<=n;j++) // 枚举第二个横坐标x2范围
            //这里定下了上下的范围，就该枚举左右的范围了
            for (int l=1,r=1,sum=0;r<=m;r++) //双指针，r为快指针
            {
                // 求解从l到r范围内，每一列的前缀和之和
                sum+=s[j][r]-s[i-1][r]; 
                while (sum>k) //此时，应该让l向r的方向靠近，同时sum减去之前l位置的和
                {
                    sum-=s[j][l]-s[i-1][l]; //减去l所在列的和
                    l++;
                }
                res+=r-l+1; //l~r范围的子矩阵都符合sum<k
        }
    printf("%lld\n",res);
    return 0;
}