历届试题 最大子阵 &P2004 领地选择（前缀和）

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

其中，A的子矩阵指在A中行和列均连续的一块。
输入格式
　　输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
　　接下来n行，每行m个整数，表示矩阵A。
输出格式
　　输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
样例说明
　　取最后一列，和为10。
数据规模和约定
　　对于50%的数据，1<=n, m<=50；
　　对于100%的数据，1<=n, m<=500，A中每个元素的绝对值不超过5000。
o（n^4）最最基础的，可以用最大子列和的思想减少一个循环。

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <map>
using namespace std;
const int inf=0x3f3f3f3f;
int a[505][505];
int main()
{
	int i,j,x,y,k,m,n,t,u,v,ans=-inf;
	scanf("%d %d",&n,&m);
/*	for(i=1;i<=m;i++)
		for(j=1;j<=n;j++){
			scanf("%d",&x);
			a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+x;
			for(u=0;u<i;u++)
				for(v=0;v<j;v++){
					ans=max(ans,a[i][j]-(a[i][v]+a[u][j]-a[u][v]));
				}
		}*/
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++){
			scanf("%d",&x);
			a[i][j]=a[i-1][j]+x;
		}
	for(i=1;i<=n;i++)
		for(j=i;j<=n;j++){
			int sum=0;
			for(k=1;k<=m;k++){
				sum+=a[j][k]-a[i-1][k];
				if(sum>ans) ans=sum;
				if(sum<0) sum=0;
			}
		}
	printf("%d",ans);
	return 0;
}

P2004 领地选择
最后一个测试点有点小坑，ans要初始化很小。。

#include<cstdio> 
#include<queue>
#include<cstring>
using namespace std;
const int maxn=500005;
int inf=0x3f3f3f3f;
int a[1005][1035]; 
int main()
{ 
	int n,m,i,j,k,t,w,x,y,ans=-inf,c;
	scanf("%d %d %d",&n,&m,&c);
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++){
			scanf("%d",&x);
			a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+x;
		}
	for(i=1;i<=n-c+1;i++){		
		for(k=1;k<=m-c+1;k++){
			int sum=a[i+c-1][k+c-1]-a[i+c-1][k-1]-a[i-1][k+c-1]+a[i-1][k-1];
		//	printf("%d %d %d\n",i,k,sum);
			if(sum>ans){
				ans=sum;
				x=i;y=k;													
			}
		}			
	}	
	printf("%d %d",x,y);
    return 0;
}