链接:https://ac.nowcoder.com/acm/contest/13493/G
来源:牛客网
这一天你来到了蓝桥杯的考场,你发现考场是一个N*M的矩阵。
因为你的群友很多,你知道考场内每个人有多强,并且把实力换算成了数值。(因为有的人太弱了,所以可能出现实力值是负数的可能)
你想知道考场内实力总和最大的矩阵区域的实力和是多少。
(注意:区域是按照矩形划分的)
输入描述:
第一行两个整数 N M \left(1\leq N \times M \leq 2e5\right)(1≤N×M≤2e5)
第二到N+1行是一个N*M的矩阵代表考场内的情况 \left( -200<=实力值<=200 \right)(−200<=实力值<=200)
输出描述:
请输出考场内实力总和最大的矩阵区域的实力和是多少
示例1
输入
复制
3 2
8 9
10 11
-4 11
输出
复制
45
示例2
输入
复制
3 2
8 9
10 11
-12 5
输出
复制
38
备注:
对于10%数据\left(1\leq N \leq M \leq 10\right)(1≤N≤M≤10)
对于40%数据\left(1\leq N \leq M \leq 100\right)(1≤N≤M≤100)
对于70%数据\left(1\leq N \leq M \leq 400\right)(1≤N≤M≤400)
对于100%数据\left(1\leq N \times M \leq 2e5\right)(1≤N×M≤2e5)
暴力枚举和朴素前缀矩阵只能拿部分分,对前缀矩阵进行压缩也只能拿70%,还需要对矩阵进行翻转以减少计算量
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
int main()
{
int n,m;
cin>>n>>m;
//相当于一个a[n+1][m+1]的数组
vector<vector<int>> a(n+1,vector<int>(m+1));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
}
}
if(n>m)//判断行长还是列长,枚举更快 ,见40行的循环 n^2 m显然让n更小最好
{
vector<vector<int>> b(m+1,vector<int>(n+1));
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
b[i][j]=a[j][i];
}
}
swap(n,m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
b[i][j]+=b[i-1][j];
}
}
ll ans=-0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
ll res=0;
for(int k=1;k<=m;k++)
{
res=max(res,0ll)+b[j][k]-b[i-1][k];
ans=max(ans,res);
}
}
}
cout<<ans<<endl;
}else
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
a[i][j]+=a[i-1][j];//计算每一列的前缀和
}
}
ll ans=-0x3f3f3f3f; //负无穷常用
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
ll res=0;
for(int k=1;k<=m;k++)
{
//把0变为ll 型,防止res退化成int
//a[j][k]-a[i-1][k]就相当于
//a[j][k],a[i-1][k],a[1][k],a[1][k-1]
//这四个顶点围成的矩形的数值
res=max(res,0ll)+a[j][k]-a[i-1][k];//求一行的最大
ans=max(ans,res);
}
}
}
cout<<ans<<endl;
}
return 0;
}