藍橋杯 歷屆試題 最大子陣

  歷屆試題 最大子陣  

時間限制：1.0s   內存限制：256.0MB

    

問題描述

　　給定一個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。

題解：最大子陣；本來是2重循環結果超時了，就改成貪心判斷ac了；

代碼：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int INF=0x3f3f3f3f;
#define mem(x,y) memset(x,y,sizeof(x))
#define SI(x) scanf("%d",&x)
#define PI(x) printf("%d",x)
#define SD(x,y) scanf("%lf%lf",&x,&y)
#define P_ printf(" ")
typedef long long LL;
const int MAXN=550;
int mp[MAXN][MAXN],dp[MAXN];
int mx;
int m,n;
int solve(int *a){
	int temp=0;
	for(int j=0;j<n;j++){
		if(temp+a[j]<a[j])temp=a[j];
		else temp+=a[j];
		mx=max(mx,temp);
	}
}
int main(){
	scanf("%d%d",&m,&n);
	mx=-INF;
	for(int i=0;i<m;i++)
		for(int j=0;j<n;j++)scanf("%d",&mp[i][j]);
	for(int i=0;i<m;i++){
		mem(dp,0);
		for(int j=i;j<m;j++){
			for(int k=0;k<n;k++)dp[k]+=mp[j][k];
			solve(dp);
		}
	}
	printf("%d\n",mx);
	return 0;
}