双指针+一维前缀和

#include <iostream>
#include<stdio.h>
using namespace std;

/* 
输入 N M K 求矩阵 nXm 最大矩阵和不超过k 
3 4 10
1 2 3 4
5 6 7 8
9 10 11 12
19  
 */

int N,M,K;
int r_sum[501][501];//存行的前缀和  
int a[501][501];
long long ans; 
int main(int argc, char *argv[]) {
	cin>>N>>M>>K;
	for(int i=1;i<=N;++i){
		for(int j=1;j<=M;++j){
			cin>>a[i][j];
			r_sum[i][j]=a[i][j]+r_sum[i][j-1];
		}
	}
	//暴力枚举两列 
	for(int col_1=1;col_1<=M;col_1++){
		for(int col_2=col_1;col_2<=M;col_2++){
			long long sum=0;
			//枚举以i为起点计算满足条件的个数 
			int rol=1;//指针一 
			//sum=r_sum[1][col_2]-r_sum[1][col_1-1];
			for(int i=1;i<=N;++i){//指针二  
				
				while(sum+r_sum[rol][col_2]-r_sum[rol][col_1-1]<=K &&
				 rol<=N){
					sum+=r_sum[rol][col_2]-r_sum[rol][col_1-1];
					rol++;
				}
				ans+=rol-i;//以i为起点连续子段小于=K的个数 为 rol-i 
				sum-= r_sum[i][col_2]-r_sum[i][col_1-1];
			}
			
		}
	} 
	printf("%lld",ans);	
	
	
	return 0;
}