(2020牛客暑期多校训练营)[二]Fake Maxpooling

题目描述
给定大小为n×m的矩阵和整数k,其中A {i,j} = lcm(i,j),是i和j的最小公倍数。 您应该找到所有大小为k×k的子矩阵中的最大值之和。
例:
(2020牛客暑期多校训练营)[二]Fake Maxpooling_第1张图片
只有一行包含三个整数n,m,k (1≤n,m≤5000, 1≤k≤min{n,m})。 只有一行包含一个整数,表示答案。
样例
3 4 2
输出
38
思路
先每一列用单调序列求出最大值,并存起来。把存起来的值每一列用单调队列用单调队列求出最大值,存起来把最后的存起来的数加起来即可。
代码
#include
using namespace std;
int a[5010][5010],gcd[5010][5010],n,m,k,i,j,h;
long long ans;
deque q;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
if(!gcd[i][j])
for(h=1;hi<=n && hj<=m;h++)
gcd[hi][hj]=h,a[hi][hj]=ijh;
for(i=1;i<=n;i++)
{
q.clear();
for(j=1;j {
while(!q.empty()&&a[i][q.front()] q.push_back(j);
}
for(j=k;j<=m;j++)
{
while(!q.empty()&&j-q.front()+1>k) q.pop_front();
while(!q.empty()&&a[i][q.front()] q.push_back(j);
a[i][j]=a[i][q.front()];
}
}
for(i=k;i<=m;i++)
{
q.clear();
for(j=1;j {
while(!q.empty()&&a[q.front()][i] q.push_back(j);
}
for(j=k;j<=n;j++)
{
while(!q.empty()&&j-q.front()+1>k) q.pop_front();
while(!q.empty()&&a[q.front()][i] q.push_back(j);
ans+=a[q.front()][i];
}
}
printf("%lld\n",ans);
}

你可能感兴趣的:((2020牛客暑期多校训练营)[二]Fake Maxpooling)