子矩阵的和
仅供个人复习使用
纯暴力会超时。枚举矩阵宽度,每次循环找宽度为j-i+1的子矩阵。再定义上下两个指针v,y。如果这段区域里的前缀和>k.把上指针下移。则对答案贡献y-v+1。
#include<bits/stdc++.h>
using namespace std;
const int N = 505;
int a[N][N];
int s[N][N];
typedef long long ll;
int main()
{
int n,m,k;
cin>>n>>m>>k;
for(int i = 1 ; i <= n ; i++)
{
for(int j = 1 ; j <= m ; j++)
{
scanf("%d",&a[i][j]);
s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j];
}
}
ll ans = 0;
for(int i = 1 ; i <= m ; i++)
{
for(int j = i ; j <= m ; j++)
{
for(int y = 1 , v = 1 ; y <= n ; y++)
{
while(v <= y && s[y][j] - s[v-1][j] - s[y][i-1] + s[v-1][i-1] > k) v++;
if(v <= y) ans += y - v + 1;
}
}
}
printf("%lld",ans);
return 0;
}