练习29，统计子矩阵【二维前缀和/双指针】

4405. 统计子矩阵 - AcWing题库

给定一个 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≤250000000。

输入样例：

3 4 10
1 2 3 4
5 6 7 8
9 10 11 12

输出样例：

19

题解：

用两个循环枚举左边界和右边界，再用前后指针枚举上下边界，复杂度为O()

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define lint long long
#define pb push_back
int n,m,k;
lint ans=0;
int a[510][510];
int main(){
    cin >> n >> m >> k;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
            a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1]; //二维前缀和
        }

    for(int i=1;i<=m;i++)
        for(int j=i;j<=m;j++){ //i枚举左边界,j枚举右边界
            for(int x=1,y=1;x<=n;x++){
                //前指针x和后指针y，分别是下边界和上边界
                while(y<=x && (a[x][j]-a[y-1][j]-a[x][i-1]+a[y-1][i-1])>k) y++;
                //边界内的子矩阵不符合条件,将y指针向下移动
                if(y<=x) ans+=x-y+1;
                //x-y+1代表从y到x的长度为1~x-y+1的子矩阵的个数
            }
        }
    cout << ans << endl;
    return 0;
}