Problem Description
给你一个m×n的整数矩阵,在上面找一个x×y的子矩阵,使子矩阵中所有元素的和最大。
Input
输入数据的第一行为一个正整数T,表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y(0< m,n<1000 AND 0< x<=m AND 0< y <=n ),表示给定的矩形有m行n列。接下来这个矩阵,有m行,每行有n个不大于1000的正整数。
Output
对于每组数据,输出一个整数,表示子矩阵的最大和。
Sample Input
1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280
Sample Output
2474
思路:和hdu1081那题差不多,只不过这题限制了子矩阵的长和宽
代码如下
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define LL long long int
using namespace std;
int a[1005][1005];
LL b[1005];
int main()
{
int T;
scanf("%d",&T);
int n,m,x,y;
while(T--)
{
memset(b,0,sizeof(b));
scanf("%d%d%d%d",&n,&m,&x,&y);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&a[i][j]);
LL ans=-1e9;
for(int j=1;j<=m;j++)
for(int i=1;i<=x;i++)
b[j]+=a[i][j];
LL sum=0;
for(int j=1;j<=m;j++)
{
if(j<y)
sum+=b[j];
else if(j==y)
{
sum+=b[j];
ans=max(ans,sum);
}
else
{
sum+=b[j]-b[j-y];
ans=max(ans,sum);
}
}
for(int i=2;i<=n+1-x;i++)
{
for(int j=1;j<=m;j++)
b[j]+=a[i+x-1][j]-a[i-1][j];
sum=0;
for(int j=1;j<=m;j++)
{
if(j<y)
sum+=b[j];
else if(j==y)
{
sum+=b[j];
ans=max(ans,sum);
}
else
{
sum+=b[j]-b[j-y];
ans=max(ans,sum);
}
}
}
printf("%lld\n",ans);
}
return 0;
}
也可以用dp来写
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define LL long long int
using namespace std;
LL dp[1005][1005];//dp[i][j]表示左上角点(1,1)和右下角点(i,j)的矩形中所有元素的和
int main()
{
int T;
scanf("%d",&T);
int n,m,x,y;
while(T--)
{
sizeof(dp,0,sizeof(dp));
LL ans=-1e9;
scanf("%d%d%d%d",&n,&m,&x,&y);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
scanf("%lld",&dp[i][j]);
dp[i][j]+=dp[i][j-1]+dp[i-1][j]-dp[i-1][j-1];
if(i>=x&&j>=y)
ans=max(ans,dp[i][j]-dp[i-x][j]-dp[i][j-y]+dp[i-x][j-y]);
}
printf("%lld\n",ans);
}
return 0;
}