题目描述

给定 𝑛×𝑛n×n 个整数组成一个方阵 𝑎𝑖,𝑗ai,j​，请找一个 𝑘×𝑘k×k 的子方阵，使得子方阵内的数字之和达到最大，输出这个最大值。

输入格式

• 第一行：两个整数 𝑛n 与 𝑘k
• 第二行到第 𝑛+1n+1 行：每行 𝑛n 个整数表示 𝑎𝑖,𝑗ai,j​

输出格式

单个整数：表示最大的 𝑘×𝑘k×k 的子方阵的数字之和。

数据范围

• 30%30% 的数据，1≤𝑘≤𝑛≤301≤k≤n≤30
• 60%60% 的数据，1≤𝑘≤𝑛≤3001≤k≤n≤300
• 100%100% 的数据，1≤𝑘≤𝑛≤25001≤k≤n≤2500
• 0≤𝑎𝑖,𝑗≤1,000,0000≤ai,j​≤1,000,000

样例数据

输入:

3 2
1 2 3
3 1 2
0 2 4

输出:

9

说明:

右下角最大

详见代码:

#include<bits/stdc++.h>
using namespace std;
int n,k;
long long a[3005][3005];
long long ans=0;
int b;
int main() 
{scanf("%d %d",&n,&k);for (int i=1;i<=n;i++){for (int j=1;j<=n;j++){scanf("%d",&b);a[i][j]=a[i][j-1]+a[i-1][j]-a[i-1][j-1]+b;}}for (int j=k;j<=n;j++){for (int i=k;i<=n;i++){ans=max(ans,a[i][j]-a[i][j-k]-a[i-k][j]+a[i-k][j-k]);}}printf("%lld",ans);return 0;
}