[蓝桥杯 2022 省 B] 统计子矩阵

[蓝桥杯 2022 省 B] 统计子矩阵

题目描述

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

输入格式

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

之后 $N$ 行每行包含 $M$ 个整数, 代表矩阵 $A$。

输出格式

一个整数代表答案。

样例 #1

样例输入 #1

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

样例输出 #1

19

提示

【样例说明】

满足条件的子矩阵一共有 $19$，包含:

大小为 $1\times 1$ 的有 $10$ 个。

大小为 $1\times 2$ 的有 $3$ 个。 大小为 $1\times 3$ 的有 $2$ 个。

大小为 $1\times 4$ 的有 $1$ 个。

大小为 $2\times 1$ 的有 $3$ 个。

【评测用例规模与约定】

对于 $30\%$ 的数据, $N,M\le 20$.

对于 $70\%$ 的数据, $N,M\le 100$.

对于 $100\%$ 的数据, $1\le N,M\le 500,0\le A_{ij}\le 1000,1\le K\le 2.5\times 10^8$.

蓝桥杯 2022 省赛 B 组 F 题。

思路

【暴力做法】

枚举四个自由度，然后用二维前缀和
求出满足<=k的所有子矩阵
但是N<=500
n^4显然会超时

【优化】

去除枚举的一个自由度
枚举上下边界后
只考虑一维的左右边界
当R确定后，L往左走，则区间和一定是单调的
此时可以利用二分和双指针
二分则时间复杂度度为n^3(不太可取)
双指针时间复杂度为n^2

【双指针】

首先预处理矩阵为一维前缀和
将L定义为R的函数
L随着R的变化而单调变化
定义：L为R确定后，能取到的最左边的值
那么R往右走，L一定向右走

【二分】

首先预处理矩阵为二维前缀和
确定上下边界后
将第三重循环定义为右边界
判断他能到达的最左边
接着进行二分即可

代码

【暴力】
通过了80%数据 n^4

//暴力做法
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510;
int n, m, k;
typedef long long ll;
int s[N][N];
ll res;

int main() {
	cin >> n >> m >> k;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			cin >> s[i][j];
			s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
		}
	for (int x1 = 1; x1 <= n; x1++)
		for (int y1 = 1; y1 <= m; y1++)
			for (int x2 = x1; x2 <= n; x2++)
				for (int y2 = y1; y2 <= m; y2++) {
					int sum = s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
					if (sum <= k)
						res++;
				}
	cout << res << endl;
	return 0;
}

【二分】
通过80%的数据 n^3

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510;
int n, m, k;
typedef long long ll;
int s[N][N];
ll res;


int sum(int x1, int y1, int x2, int y2) {
	int p = 0;
	p += s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
	return p;
}

int main() {
	cin >> n >> m >> k;
	//读入矩阵的值,并预处理二维前缀和
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			cin >> s[i][j];
			s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] ;
		}
	ll res = 0;
	//枚举上下边界，利用二分求出答案
	for (int i = 1; i <= n; i++)
		for (int j = i; j <= n; j++)
			for (int t = 1; t <= m; t++) {
				int l = 1, r = t;
				while (l < r) {
					int mid = l + r >> 1;
					if (sum(i, mid, j, t) > k)
						l = mid + 1;
					else
						r = mid;
				}
				if (sum(i, l, j, t) <= k)
					res += t - l + 1;
			}
	cout << res << endl;
	return 0;
}

【双指针】
通过了100%数据 n^2

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510;
int n, m, k;
typedef long long ll;
int s[N][N];

int main() {
	cin >> n >> m >> k;
	//读入矩阵的值,并预处理一维前缀和
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			cin >> s[i][j];
			s[i][j] += s[i - 1][j];
		}
	//枚举上下边界，并利用双指针求出答案
	ll res = 0;
	for (int i = 1; i <= n; i++)
		for (int j = i; j <= n; j++)
			for (int l = 1, r = 1, sum = 0; r <= m; r++) {
				sum += s[j][r] - s[i - 1][r];
				while (sum > k) {
					sum -= s[j][l] - s[i - 1][l];
					l++;
				}
				res += r - l + 1;
			}
	cout << res << endl;
	return 0;

}