蓝桥杯原题——统计子矩阵（双指针 + 前缀和）图示+题解

题目：

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

输入格式

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

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

输出格式

一个整数代表答案。

数据范围

对于 30%30% 的数据，N,M≤20，
对于 70%70% 的数据，N,M≤100，
对于 100%100% 的数据，1≤N,M≤500;0≤Aij≤1000;1≤K≤2.5×10^8

输入样例：

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

输出样例：

19

样例解释

满足条件的子矩阵一共有 1919，包含：

• 大小为 1×11×1 的有 1010 个。
• 大小为 1×21×2 的有 33 个。
• 大小为 1×31×3 的有 22 个。
• 大小为 1×41×4 的有 11 个。
• 大小为 2×12×1 的有 33 个。

 解题思路：

求一个区间内所有数的和可以用前缀和优化掉两重累加元素的for循环

分析题意可知，如果搜索到的一个矩阵满足题目要求（其内部元素之和<=k）那么它的子矩阵都满足要求，我们可以只枚举上下区间，左右区间用双指针动态维护一个恰好满足题目要求的区间即可，规定右指针为快指针，左指针为慢指针。注意这里的恰好：左指针一定是处于边界处即左指针往左再往左包含一个就不满足题目要求了，而当前“恰好”满足要求，这样才能做到枚举时不重不漏，这样又可以优化掉一层for循环，达到了题目要求。

代码相关图示：

有注释的代码： 

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

using namespace std;

const int N = 510;
int matrix[N][N];
int sum[N][N];
int n, m, k;

int main(){
	cin >> n >> m >> k; 
	for(int i = 1; i <= n; i++){//数据输入 
		for(int j = 1; j <= m; j++){
			cin >> matrix[i][j];
		}
	}
	for(int i = 1; i <= n; i++){//容斥原理求二维前缀和数组 
		for(int j = 1; j <= m; j++){
			sum[i][j] = matrix[i][j] + sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1];
		}
	}

	long long cnt = 0;
	for(int x1 = 1; x1 <= n; x1++){//枚举上区间 
		for(int x2 = x1; x2 <= n; x2++){//枚举下区间 
			for(int y1 = 1, y2 = 1; y2 <= m; y2++){//左右区间用前缀和进行优化 
				while(sum[x2][y2] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1] > k){
					y1++;//如果区间和比k大则左指针（慢指针）需要右移保证区间和满足题目要求 
				}
				cnt += y2 - y1 + 1;//双指针保证整个左右区间中的结果都符合题意，即区间和都小于k
			}//右指针每移动一次都会产生新的一组符合题意的解（因为搜索到的每组解都以右指针为右边界，右指针从前往后遍历，不会遍历重复解） 
		}//将用于计数的cnt每次加上左右区间中所有解的个数即可 
	}
	cout << cnt << endl;//下面是O(n^4)的纯暴力做法 
//	int res = 0;
//	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++){
//					if(sum[x2][y2] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1] <= k){
//						res++;
//					}
//				}
//			}
//		}
//	}
//	cout << res << endl;
	return 0;
};