单调队列 2020牛客暑期多校训练营（第二场）F题 Fake Maxpooling

题目大意：

给你一个 n* m 的网格，每个格子有一个元素 A(i,j)，这个元素 A(i,j) 等于 lcm(i,j)；问所有 k* k 的正方形的最大值（最大值是指 k*k 正方形里面的元素最大值）之和；

---

比赛时暴力找规律过了，正解是用单调队列求出每一行的 m-k+1 个最大值，然后在这个新求得的网格中再用单调队列求出每一列的 n-k+1 个最大值，最后相加就是答案，注意 longlong ;

代码：

#include
#define LL long long
#define pa pair
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=5100;
const int M=2000100;
const LL mod=1e9+7;
int a[N][N],b[N][N],sum[N][N];
int head,tail,cnt;
struct Node{
	int post,date;
}que[N];
void init(){
	head=tail=0;
	for(int i=0;i<N;i++) que[i].date=que[i].post=0; 
}
int main(){
	int n,m,k;cin>>n>>m>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++) a[i][j]=(i*j)/(__gcd(i,j));
	}
	for(int i=1;i<=n;i++){
		init(),cnt=0;
		for(int j=1;j<=m;j++){
			while(tail&&head<=tail&&que[tail].date<=a[i][j]) tail--;
			que[++tail].date=a[i][j],que[tail].post=j;
			while(head<=tail&&que[head].post<=j-k) head++;
			if(j>=k) b[i][++cnt]=que[head].date;
		}
	}
	LL ans=0;
	for(int i=1;i<=cnt;i++){
		init();
		for(int j=1;j<=n;j++){
			while(tail&&head<=tail&&que[tail].date<=b[j][i]) tail--;
			que[++tail].date=b[j][i],que[tail].post=j;
			while(head<=tail&&que[head].post<=j-k) head++;
			if(j>=k) ans+=(LL)que[head].date;
		}
	}
	cout<<ans<<endl;
	return 0;
}