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

实现最大池化
其实就是单调队列的滑动窗口,从一维到二维只是多求了一遍而已。
滑动窗口算法的关键在于用双端队列维护当前窗口内的值和位置,保持值单调,队首位置在窗口以外的移出队列,纸样就可以保证队首一直是当前窗口内的最值,复杂度O(mn)。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn=5e3+10;
int mat[maxn][maxn];
int mat_second[maxn][maxn];

int n,m,k;
int gcd(int x,int y){
    return y?gcd(y,x%y):x;
}

int lcm(int x,int y){
    return x*y/gcd(x,y);
}

deque<pair<int,int> >dq;

int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            mat[i][j]=lcm(i,j);
        }
    }
    for(int i=1;i<=n;i++){
        while(!dq.empty())dq.pop_front();
        for(int j=1;j<=k;j++){
            int x=mat[i][j];
            while(!dq.empty()&&dq.back().second<=x){
                dq.pop_back();
            }
            dq.push_back({j,x});
        }
        mat_second[i][k]=dq.front().second;
        for(int j=k+1;j<=m;j++){
            int x=mat[i][j];
            while(!dq.empty()&&dq.back().second<=x){
                dq.pop_back();
            }
            dq.push_back({j,x});
            while(!dq.empty()&&j-dq.front().first>=k)
                dq.pop_front();
            mat_second[i][j]=dq.front().second;
        }
    }
    ll ans=0;
    for(int j=k;j<=m;j++){
        while(!dq.empty())dq.pop_front();
        for(int i=1;i<=k;i++){
            int x=mat_second[i][j];
            while(!dq.empty()&&dq.back().second<=x){
                dq.pop_back();
            }
            dq.push_back({i,x});
        }
        ans+=dq.front().second;
        for(int i=k+1;i<=n;i++){
            int x=mat_second[i][j];
            while(!dq.empty()&&dq.back().second<=x){
                dq.pop_back();
            }
            dq.push_back({i,x});
            while(!dq.empty()&&i-dq.front().first>=k)
                dq.pop_front();
            ans+=dq.front().second;
        }
    }
    printf("%lld\n",ans);
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值