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

 

 本题所用到的算法:二维单调队列

一维单调队列:

 首先不难分析窗口是这样滑动的:

 

如果我们使用尺取/滑动窗口,时间复杂度为O(n*k),当k很大时容易超时。本题采用单调队列优化

所谓单调队列即双端队列,队列中的值是单调的,在每次滑动之后维持队列的单调性

对于本题来说,每次都是最左端的数移除队列,最右边的数加入队列,由于队列的单调性我们直接取队列的最左端的值,即为区间最值

对于此题来说需要降序的单调队列:

对于新加入的数,我们需要判断前面是否都是比它大的。因为如果它更大,区间在移动过程中(实际是该数慢慢左移直到移出队列过程中),前面的数不可能作为最大值,所以我们需要降序维持队列单调性

 

队列中直接存每个数吗?

  • 一开始我也是这么想的,但是写着写着发现无法实现,如果我们存数的话我们需要知道当前队列的长度是否超过k。但是区间长度是变化的,一般小于k,因此没有办法动态维护

  • 因此我们存每个数的下标去操作,具体看下面的常规代码

常规代码:

#include <set>
#include <map>
#include <stack>
#include <queue>
#include<iostream>
using namespace std;
#define lowbit(x) (x&(-x))
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> P;
const double eps=1e-8;
const int inf=0x3f3f3f3f;
const ll INF=1e18;
const int Mod=1e9+7;
const int maxn=2e5+10;

int q[maxn],a[maxn];  //数组维护双端队列即可

int main(){
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    int l=1,r=1;
    for(int i=1;i<=n;i++){   //注意边界的判断
        while(r-l>0 && q[l]<=i-k) 
            l++;  //将前面超过长度k的删除
        while(r-l>0 && a[q[r-1]]<=a[i])
            r--;  //将小于这个下标值删除,注意这里是a[q[r-1]]
        q[r++]=i;  //每次添加当前下标
        if(i>=k) 
            cout<<a[q[l]]<<" "; //只有长度达到k之后才能输出区间最值
    }
    return 0;
}
//8 3
//2 3 4 2 6 2 5 1

本题:

思路:

    考虑 一维 求每一行每k 个元素的最大值 ,很容易想到单调队列(滑动窗口),可以维护每个长度为 k 的区间的最大值,但这里是二维的,其实只要在一维的基础上 再对列做单调队列。因为在一维单调队列的基础上,我们已经求得了第 i 行 第 j个元素为起点长度为 k 的区间的最大值 ,在求整个区间时就只要用我们已经得到的值来求就好了。

#include<iostream>
#include<stdio.h>
using namespace std;
typedef long long ll;
const int maxn = 5e3+5;
int a[maxn][maxn],ans[maxn][maxn],que[maxn];
int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
int lcm(int a,int b){
    return a*b/gcd(a,b);
}

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]=lcm(i,j);
        }
    }
    for(int i=1;i<=n;i++){
        int head=1,tail=1;
        for(int j=1;j<=m;j++){
            while(tail-head>0&&que[head]<=j-k) head++;
            while(tail-head>0&&a[i][que[tail-1]]<=a[i][j]) tail--;
            que[tail++]=j;
            ans[i][j] = a[i][que[head]];
//            cout<<ans[i][j]<<endl;
        }
    }
    for(int j=1;j<=m;j++){
        int head=1,tail=1;
        for(int i=1;i<=n;i++){
            while(tail-head>0&&que[head]<=i-k) head++;
            while(tail-head>0&&ans[que[tail-1]][j]<=ans[i][j]) tail--;
            que[tail++] =i;
            ans[i][j] = ans[que[head]][j];
//            cout<<ans[i][j]<<endl;
        }
    }
    ll anss=0;
    for(int i=k;i<=n;i++){
        for(int j=k;j<=m;j++){
            anss+=ans[i][j];
//            cout<<ans[i][j]<<" ";
        }
//        cout<<endl;
    }
    cout<<anss<<endl;
    return 0;
}

 

posted @ 2020-07-14 15:25  sqsq  阅读(224)  评论(0编辑  收藏  举报