根据数据范围n,m<=500 知 复杂度应控制在n的三次方。
两个自由度控制子矩阵的上下边界。
此时只需要对列求和——> 一位前缀和 (在能用一维前缀和的情况下不用二维,一维更快)
只需控制左右边界,这里用双指针 l = f ( r ) 在r确定的情况下,f ( r )为满足条件的最左边的l ,此情况下(r确定的情况下)这样就有r-l+1个子矩阵。
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=510;
int a[N][N];
int n,m,k;
signed main()
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
a[i][j]+=a[i-1][j]; //每列的前缀和 a[i][j]表示 第j列前i个数的和
}
int res=0;
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+=a[j][r]-a[i-1][r];
while(sum>k)//超过了就让l往右挪动
{
sum-=a[j][l]-a[i-1][l];
l++;
}
res+=r-l+1;
}
}
cout<<res;
}