蓝桥杯2022年第十三届省赛真题-统计子矩阵

题干 

时间限制: 1s 内存限制: 256MB 提交: 5532 解决: 1141

题目描述

给定一个 N × M 的矩阵 A，请你统计有多少个子矩阵 (最小 1 × 1，最大 N × M) 满足子矩阵中所有数的和不超过给定的整数 K?

输入

第一行包含三个整数 N, M 和 K.

之后 N 行每行包含 M 个整数，代表矩阵 A.

输出

一个整数代表答案。

样例输入

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 ≤ A[i][j] ≤ 1000; 1 ≤ K ≤ 250000000.

题意分析 

        在大矩阵中寻找所有数和小于k的子矩阵数量，子矩阵的和可以让我们很轻松的想到二维前缀和，但是直接二维前缀和然后暴力判断所有情况的话时间复杂度到了O(N^4)，对于500的数量级的话会超时，但是题目中给出了元素非负的条件，那么我们就可以利用前缀和的单调递增性，去减少一层循环，（如果一个子矩阵的和已经大于k，那么比这个子矩阵更大的子矩阵一定大于k），具体代码如下

#include<iostream>
using namespace std;

typedef long long LL;
const int N = 510;

int a[N][N],s[N][N];
LL sum,k;
int n,m;
int main(){
	cin >> n >> m >> k;
	for(int i = 1;i <= n;i++){
		for(int j = 1;j <= m;j++){
			cin >> a[i][j];
			s[i][j] = a[i][j] - s[i-1][j-1] + s[i-1][j] + s[i][j-1];
		}
	}
	
	for(int i = 1;i <= n;i++){
		for(int j = i;j <= n;j++){
			for(int l = 1,r = 1;r <= m;r++){
				从s[i][l] 到 s[j][r] 的和大于k
				且 l 在 r 的上方
				对于每次最里层循环 i j r 是固定的 寻找对应的 l 则可得有 r - l + 1 个子矩阵是满足条件的
				while(l <= r && (s[j][r] - s[j][l-1] - s[i-1][r] + s[i-1][l-1]) > k){
					++l;
				}
				sum += r - l + 1;
			}
		}
	}
	cout << sum; 
}

总结

        题干非常清晰的一个二维前缀和模板题，只需要些许数学优化即可，适合新手练习二维前缀和技巧。 

4405. 统计子矩阵 - AcWing题库