第十三届蓝桥杯大赛软件赛省赛C/C++ 大学 B 组 统计子矩阵

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<queue>

using namespace std;

int cnt,temp;
int n,m,K;
int a[505][505];
int pre[505][505];//二维前缀和



void sol()
{
	cin>>n>>m>>K;
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
		{
			cin>>a[i][j];
			pre[i][j]=a[i][j]+pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1];
		}
	}
	int limit;//等到第四层循环时，在k行l列时超过K,记录下其l值。当i，j不变，k为k+1行时，再次达到l时一定会再次超过K，所以进行剪枝
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=m;++j)
		{
			limit=m;
			for(int k=i;k<=n;++k)
			{
				for(int l=j;l<=m&&l<=limit;++l)
				{
					temp=pre[k][l]-pre[i-1][l]-pre[k][j-1]+pre[i-1][j-1];
					if(temp<=K)++cnt;
					else {
						limit=l;
						break;
					}
				}
			}
		}
	}
	
    cout<<cnt<<endl;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	sol();
}