题面描述
求一个从
n
∗
n
n*n
n∗n的矩阵中选出权值和最大的
r
∗
r
r*r
r∗r的子矩阵。
思路
注意:此题居然卡空间!
就相当一个二维前缀和
另外,有
n
<
r
n<r
n<r的情况
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=5e3+10;
int f[N][N];int x,y,z;
int main()
{
int n,R;scanf("%d%d",&n,&R);
int r,c;r=c=R;
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&x,&y,&z);
x++,y++;f[x][y]=z;
r=max(r,x);c=max(c,y);
}
f[0][0]=f[0][1]=f[1][0]=0;
for(int i=1;i<=r;i++)
for(int j=1;j<=c;j++)
f[i][j]=f[i-1][j]+f[i][j-1]-f[i-1][j-1]+f[i][j];
int ans=0;
for(int i=R;i<=r;i++)
for(int j=R;j<=c;j++)
ans=max(ans,f[i][j]+f[i-R][j-R]-f[i][j-R]-f[i-R][j]);
printf("%d\n",ans);
return 0;
}