问题 1445: [蓝桥杯][历届试题]最大子阵

题目描述
给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。

其中，A的子矩阵指在A中行和列均连续的一块。

样例说明
取最后一列，和为10。
数据规模和约定
对于100%的数据，1< =n, m< =500，A中每个元素的绝对值不超过5000。

输入
输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
接下来n行，每行m个整数，表示矩阵A。
输出
输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10

这道题其实是二维的最大字段和的问题，我们只需要把上面的每一列压缩下来，枚举每一行，二维变一维就好做了。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const	int N=510;
int	a[N][N];
int sc[N];
int n,m; 
int get_ans(int a[])
{
	int ans=0;
	int maxv=-1e9;
	int dp[N];
	dp[0]=a[0];
	for(int i=1;i<m;i++)
	{
		dp[i]=max(a[i],a[i]+dp[i-1]);
		maxv=max(dp[i],maxv);
	}
	return maxv;
}
int main()
{
	cin>>n>>m;
	int ans=-1e9;
	for(int i=0;i<n;i++)
	for(int j=0;j<m;j++)
	cin>>a[i][j];
	for(int i=0;i<n;i++)
	{
		memset(sc,0,sizeof sc);
	for(int j=i;j>=0;j--)
	{
	for(int k=0;k<m;k++)
	{
		sc[k]+=a[j][k];
	
	}
	ans=max(ans,get_ans(sc));
	}	
	}
	cout<<ans<<endl;
    return 0;
}