POJ 1050 To the max(最大子矩阵)

问题简述

给一个n*n矩阵(n<=100),求出该矩阵的元素和最大得子矩阵元素和

解决方法

这里要用到动态规划的一个经典模型:最大子矩阵

用s[i][j]表示前i行的第j列的数的总和，则s[j][k]-s[i-1][k]表示第i-j行的第k列的元素总和

用f[i][j][k]表示第i行到第j行，以第k列数结尾的最大子矩阵的和，则当每列的元素和可以在O（1）时间内计算出来时，可将该模型简化为最大连续和子序列

f[i][j][k]=max{0,f[i][j][k-1]}+s[j][k]-s[i-1][k]其中0表示只选择第k个元素为子序列，f[i][j][k-1]表示将第k个元素和以第k-1个元素结尾的子序列拼在一起

#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=105;
const int inf=0x3f3f3f3f;
int s[maxn][maxn],n,f[maxn][maxn][maxn];
int main(){
	int x,y,t,ans=-inf;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			scanf("%d",&x);
			s[i][j]=s[i-1][j]+x;
		}
	for(int i=1;i<=n;i++)
		for(int j=i;j<=n;j++)
			for(int k=1;k<=n;k++)f[i][j][k]=-inf;
//由于元素可以为负故初始化为无穷小
	for(int i=1;i<=n;i++)
		for(int j=i;j<=n;j++){
			f[i][j][1]=s[j][1]-s[i-1][1];
			ans=max(ans,f[i][j][1]);
			for(int k=2;k<=n;k++){
				f[i][j][k]=max(f[i][j][k-1],0)+s[j][k]-s[i-1][k];
				ans=max(ans,f[i][j][k]);
			}
		}
	cout<<ans;
}