【hiho一下_week260】最大子矩阵

hihocoder的hiho一下，第260周的题目

---

• 原题
• 我的思路
• 解析

---

原题

传送门：https://hihocoder.com/contest/hiho260/problem/1

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

给定一个NxM的矩阵A和一个整数K，小Hi希望你能求出其中最大（元素数目最多）的子矩阵，并且该子矩阵中所有元素的和不超过K。

输入

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

以下N行每行包含M个整数，表示A。

对于40%的数据，1 <= N, M <= 10

对于100%的数据，1 <= N, M <= 250 1 <= K <= 2147483647 1 <= Aij <= 10000

输出

满足条件最大的子矩阵所包含的元素数目。如果没有子矩阵满足条件，输出-1。

---

我的思路

最先想到的还是暴力，先把每个格子到左上角格子所组成矩阵的和计算出来，然后按行列遍历各种情况进行检索把每个子矩阵的和计算，找出不超过规定数字的最大和子矩阵所包含的元素数目。

代码：

#include <iostream>
#include <algorithm>
#include <memory.h>

using namespace std;


int main(int argc, char** argv) {
	int n, m, k;
	int ans = 0;
	int a[255][255];
	memset(a, 0, sizeof(a));
	
	cin >> n >> m >> k; 
	
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			cin >> a[i][j];
		}
	}
	
	// 先把第一行和第一列的计算了，后面就可以依次按公式计算了 
	for (int i = 2; i <= n; i++) {
		a[1][i] = a[1][i-1] + a[1][i];
	}
	for (int i = 2; i <= m; i++) {
		a[i][1] = a[i-1][1] + a[i][1]; 
	}
	
	for (int i = 2; i <= n; i++) {
		for (int j = 2; j <= m; j++) {
			a[i][j] = a[i-1][j] + a[i][j-1] - a[i-1][j-1] + a[i][j]; 
		}
	}
	
	for(int i=1; i<=n; i++) {
		for(int j=i; j<=n; j++)  //枚举行
		{
			for(int q=1; q<=m; q++)
				for(int p=1; p<=q; p++)
				{
					int sum = a[j][q]  + a[i-1][p-1] - a[j][p-1] - a[i-1][q];  //矩阵 i, j, p, q
					if (sum <= k)
					{
						ans = max(ans, (j-i+1)*(q-p+1));
						break;
					}
				}
		}
	}
		
	cout<<ans;

	return 0;
}

---

解析

• 解析

本题是一道经典的面试题目。

首先我们可以枚举矩阵的左边界和右边界，不妨设为L和R。这一步枚举的时间复杂度是O(N^2)。

一旦确定L和R，问题就2维降到了1维。我们令

B[0] = A[0][L] + A[0][L+1] + … + A[0][R] B[1] = A[1][L] + A[1][L+1]

• … + A[1][R] … B[N-1] = A[N-1][L] + B[N-1][L+1] + … + B[N-1][R] 提示：经过O(N^2)的预处理前缀和之后，B数组的每一项可以O(1)求得。这里不再赘述。

原问题就转化为求B中最长的连续子数组，满足子数组的和不超过K。

注意到矩阵A中的元素都是正整数，所以B数组中的元素也都是正整数。于是我们就可以用双指针(也被称为滑动窗口)的方法解决B数组的问题：

初始时i=j=0

只要B[i] + … + B[j]的和小于等于K就令j累加1;否则令i累加1。

直到i==N。在此过程中，j-i的最大值就代表了最长的子数组。 当然实现的时候还有一些边界问题要处理，我们就不赘述了。