统计子矩阵

十三届蓝桥杯C++B组第F题，求各个子矩阵和

样例输入：

3 4 10
1 2 3 4
5 6 7 8
9 10 11 12

样例输出：

19

【样例说明】

满足条件的子矩阵一共有 19，

包含： 大小为 1 × 1 的有 10 个。

大小为 1 × 2 的有 3 个。

大小为 1 × 3 的有 2 个。

大小为 1 × 4 的有 1 个。

大小为 2 × 1 的有 3 个。

【评测用例规模与约定】

对于 30% 的数据，N, M ≤ 20.

对于 70% 的数据，N, M ≤ 100.

对于 100% 的数据，1 ≤ N, M ≤ 500; 0 ≤ Ai j ≤ 1000; 1 ≤ K ≤ 250000000.

解：

常规的二维前缀和+计算子矩阵和是较为简易的方法求解，这需要两层for定位矩阵左上角和两层for定位矩阵右下角，一共用了O(N^4)，题目对百分百数据是500^4有6.25*10^10的数据量，无疑是会报时间的，但是也能拿个十来分的分数了🥳，我比赛的时候还感慨10的六次呢，能有5分就不戳了。

此时我们可以对四次的解法进行优化，把二维前缀改成竖直放向上的前缀和，用两层for确定子矩阵的上下范围，其中第三层for来放子矩阵的左右范围，用l控左边，r控右边且在for内自加，计算子矩阵的和，如果和sum>k，那么就把子矩阵左边界上的值给减了，同时把左边界l右移，当r==m的时候，当左右子矩阵都满足条件时，l=1，这一层的满足子矩阵就有r-l+1，以此可得，每当r确定一个值得时候就有r-l+1种情况满足，此时的时间复杂度比O(N^3)多一点，优化了原来的O(N^4)

附上代码：🥰

#include<iostream>
using namespace std;
const int N = 10010;
int n,m,k,a[N][N];
int 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];
        }
    }
    long long ans=0,l,r;
    for(int i=1;i<=n;++i){
		for(int j=0;j<i;++j){
			int sum=0;
			for(l=1,r=1;r<=m;++r){
				sum+=a[i][r]-a[j][r];
				while(sum>k){
					sum-=a[i][l]-a[j][l];
					l++;
				}
			if(r>=l)ans+=r-l+1;
			}
		}
	}
	cout<<ans;
}