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

题目描述

给定一个 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 ≤ Ai j ≤ 1000; 1 ≤ K ≤ 250000000.

总结：

二位前缀法：定义s[i][j]表示s[1][1]-s[i][j]的和

 

尺取法：一维区间和

思想：

1.先求每行的前缀和

2.先找大矩阵，然后大矩阵的每行都是一个小矩阵

#include <iostream>
using namespace std;
const long long  MOD = 1000000007;
const int maxsize = 1000;

long long m, n, k;
long long a[maxsize][maxsize]={0};
long long ans = 0;

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][j - 1];//先对于每一行求前缀和
		}
	}
	for (int i = 1; i <= m; i++) {//左边界
		for (int j = i; j <= m; j++) {//右边界
			int top = 1;//上边界
			long long cur_sum = 0;//目前的子矩阵和
			for (int down = 1; down <= n; down++) {//下边界
				cur_sum += a[down][j] - a[down][i - 1];//下边界前缀和
				while (cur_sum > k) {//如果下边界前缀和大于k
                    //从top行开始减
					cur_sum -= a[top][j] - a[top][i - 1];
					top++;
				}
				ans += down - top + 1;
			}
		}
	}
	cout << ans << endl;
	return 0;
}