子矩阵的和

子矩阵的和
仅供个人复习使用

纯暴力会超时。枚举矩阵宽度，每次循环找宽度为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;
}