CodeForces - 1060C(前缀和+思维)

题意:

有一个长度为 n 的 a 数组和长度为 m 的 b 数组,让这个一维矩阵相乘得到二维的矩阵,求最大子矩阵的和小于或等于 x。

题解:

有一道原题,区间范围是1000,时间复杂度是O(n^3),用在这里会超时。(Hihocoder 1502: 最大子矩阵)

我们可以发现一个规律:一个子矩阵的和就是 (a[i]+a[i+1]+...+a[j])*(b[i]+b[i+1]+...+b[j])。

所以我们可以使用前缀和求一个子矩阵的和,我们要求的是子矩阵的大小,所以设一个数组存长度为 1...n 的 a 和 b 的前缀和,再遍历求最大的大小。

#include 
#include  
#include   
#include   
#include    
#include    
#include    
#include    
#include     
#include     
#include     
#include     
#include      
#include       
#include       
#define line printf("---------------------------\n") 
#define mem(a, b) memset(a, b, sizeof(a))
#define pi acos(-1)
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;
const int maxn = 2000+10;

ll a[maxn], b[maxn], ma[maxn], mb[maxn];

int main(){
	int n, m;
	scanf("%d %d", &n, &m);
	a[0] = b[0] = 0;
	for(int i = 1; i <= n; i++){
		scanf("%d", &a[i]);
		a[i] += a[i-1];
	}
	for(int i = 1; i <= m; i++){
		scanf("%d", &b[i]);
		b[i] += b[i-1];
	}
	mem(ma, inf);
	mem(mb, inf);
	for(int i = 1; i <= n; i++){
		for(int j = i; j <= n; j++){
			ma[j-i+1] = min(ma[j-i+1], a[j]-a[i-1]);
		}
	}
	for(int i = 1; i <= m; i++){
		for(int j = i; j <= m; j++){
			mb[j-i+1] = min(mb[j-i+1], b[j]-b[i-1]);
		}
	}
	ll x;
	scanf("%lld", &x);
	int ans = 0;
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			if(ma[i]*mb[j] <= x){
				ans = max(ans, i*j);
			}
		}
	}
	printf("%d\n", ans);
}

 

你可能感兴趣的:(codeforces题解)