POJ 1050

题意：出一个矩阵（100×100）求一个子矩阵，使得子矩阵中各个元素的和最大。

思路：一维数组最大连续字段和的变形，具体思路见代码注释。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<stack>
#include<algorithm>
#include<vector>
#define y1 y12345
#define mx 110
#define inf 0x3f3f3f3f
#define LL long long
#define ULL unsigned long long
#define mod 1000000007
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;

int a[mx][mx];
int s[mx][mx];       //sum[i][j]记录sum[i][1]到sum[i][j]的和
int dp[mx][mx][mx];  //dp[i][j][k] 第i行到第j行中第1列到第k列的最大子矩阵和
int max(int a,int b){
    if(a>b)return a;
    return b;
}

int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%d",&a[i][j]);
    
    for(int j=1;j<=n;j++)
        for(int i=1;i<=n;i++)
            s[i][j]=s[i-1][j]+a[i][j];
    
    memset(dp,0,sizeof(dp));
    int cmax=-inf;
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            int sum=0;
            for(int k=1;k<=n;k++){                       //将第i行到第j行的一列当做以一个元素求最大连续子段和
                int now=s[j][k]-s[i-1][k];               //第i行到第j行中第k列的和
                dp[i][j][k]=max(dp[i][j][k-1]+now,now);  
                cmax=max(cmax,dp[i][j][k]);
            }
        }
    }
    printf("%d\n",cmax);
    return 0;
}