统计子矩阵（二维前缀和，滑动窗口，双指针）

问题描述

给定一个 N×M 的矩阵 A, 请你统计有多少个子矩阵 (最小 1×1 最大 N×M) 满足子矩阵中所有数的和不超过给定的整数 K ?

输入格式

第一行包含三个整数 N,M 和 K.

之后 N 行每行包含 M 个整数, 代表矩阵 A.

输出格式

一个整数代表答案。

样例输入

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 个。

评测用例规模与约定

对于 30% 的数据, N,M≤20.

对于 70% 的数据, N,M≤100.

对于 100% 的数据, 1≤N,M≤500;0≤Aij​≤1000;1≤K≤250000000.

 

70分思路：

遍历所有子阵，用b数组储存 列方向的前缀和，将N*M矩阵压缩为1*M的一维矩阵，sum数组储存一维矩阵的前缀和，然后通过l和r暴力搜索所有可能的和，得70分

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[505][505],b[505][505],sum[505];
int main(){
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>a[i][j];
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++){
        memset(b,0,sizeof(b));
        for(int j=i;j<=n;j++){
            for(int k=1;k<=m;k++){
                b[j][k]=b[j-1][k]+a[j][k];
            }
            memset(sum,0,sizeof(sum));
            for(int k=1;k<=m;k++){
                sum[k]=sum[k-1]+b[j][k];
            }
            for(int l=0;l<m;l++){
                for(int r=l+1;r<=m;r++){
                    if(sum[r]-sum[l]<=k)ans++;
                }
            }
        }
    }
    cout<<ans;
    return 0;
}

AC思路：依旧用b数组储存 列方向前缀和，使用双指针对内部优化，维持一个窗口，窗口内元素的个数即为右指针当前位置符合条件的矩阵数，窗口右指针不断右移，sum储存窗口的值，sum>k时，左指针右移，维护sum始终<=k，即可AC

​
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[505][505],b[505][505];   //sum[505]
int main(){
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>a[i][j];
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++){
        memset(b,0,sizeof(b));
        for(int j=i;j<=n;j++){
            for(int k=1;k<=m;k++){
                b[j][k]=b[j-1][k]+a[j][k];
            }
            /*memset(sum,0,sizeof(sum));
            for(int k=1;k<=m;k++){
                sum[k]=sum[k-1]+b[j][k];
            }
            for(int r=m;r>=1;r--){    //90分代码
                int l=0;
                while(sum[r]>k){        //暴搜剪枝，多过20分
                    l++;
                    sum[r]-=b[j][l];
                    if(l==r)break;
                }
                ans+=r-l;
            }*/
            for(int l=1,r=1,sum=0;r<=m;r++){
				sum+=b[j][r];           //双指针优化，移动右指针
				while(sum>k){           //维护一个始终小于k的滑动窗口，成功AC
				    sum-=b[j][l];
				    l++;
				}
				ans+=r-l+1;
            }
        }
    }
    cout<<ans;
    return 0;
}

​