Fake Maxpooling
题目描述:
给定大小为n×m的矩阵和整数k,其中A {i,j} = lcm(i,j),是i和j的最小公倍数。 您应该找到所有大小为k×k的子矩阵中的最大值之和
输入:
输入只有一行:三个整数n,m,k(1≤n,m≤5000,1≤k≤min{n,m})
输出:
一个整数ans,为所有大小为k×k的子矩阵中的最大值之和
样例输入:
3 4 2
样例输出:
38
思路:单调队列+滑动窗口。
#include<cstdio>
#include<deque>
using namespace std;
const int MAXN=5005;
int a[MAXN][MAXN],gcd[MAXN][MAXN];
long long ans;
deque<int> q;
int n,m,k;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(!gcd[i][j])
for(int z=1;z*i<=n&&z*j<=m;z++)
{
gcd[z*i][z*j]=z;
a[z*i][z*j]=i*j*z;
}//利用类似于素数筛的方法建图,时间复杂度O(mn)
for(int i=1;i<=n;i++)
{
while(!q.empty()) q.pop_back();
for(int j=1;j<k;j++)
{
while(q.size()&&a[i][q.front()]<a[i][j]) q.pop_back();
q.push_back(j);
}
for(int j=k;j<=m;j++)
{
while(q.size()&&q.front()<j-k+1) q.pop_front();
while(q.size()&&a[i][q.front()]<a[i][j]) q.pop_back();
q.push_back(j);
a[i][j]=a[i][q.front()];
}
}
for(int i=k;i<=m;i++)
{
while(!q.empty()) q.pop_back();
for(int j=1;j<k;j++)
{
while(q.size()&&a[q.front()][i]<a[j][i]) q.pop_back();
q.push_back(j);
}
for(int j=k;j<=n;j++)
{
while(q.size()&&q.front()<j-k+1) q.pop_front();
while(q.size()&&a[q.front()][i]<a[j][i]) q.pop_back();
q.push_back(j);
ans+=a[q.front()][i];
}
}
printf("%lld\n",ans);
}