[POJ2823]Sliding Window(单调队列)

题目链接:

http://poj.org/problem?id=2823

题意:

给你一个序列a，要找到两个序列
a的下标为i,i+1,i+2, ….. i+k-1 当中的最小值为b[i];
c[i]就是最大值

题解:

单调队列，紫书中第八章的滑动窗口。。

优先队列也可做

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e6+10;

struct node{
    int x,y;
}v[maxn];

int n,k;
int a[maxn],mi[maxn],mx[maxn];

void getmin(){
    int head=1,tail=0;
    for(int i=1; i<k; i++){
        while(head<=tail && a[i]<=v[tail].x) tail--; 
        tail++; v[tail].x=a[i],v[tail].y=i;
    }
    for(int i=k; i<=n; i++){
        while(head<=tail && a[i]<=v[tail].x) tail--; // 插入到第一个比a[i]小的下一个位置，并把无用的【不可能出现在答案里的值，如果后面的一个位置比这个位置的值小，那么肯定要小的】删除
        tail++; v[tail].x=a[i],v[tail].y=i;
        while(head<=tail && i-k+1>v[head].y) head++; // 删除队首，已经不再这个窗口里的元素
        mi[i-k+1] = v[head].x;
    }
}

void getmax(){
    int head=1,tail=0;
    for(int i=1; i<k; i++){
        while(head<=tail && a[i]>=v[tail].x) tail--;
        tail++; v[tail].x=a[i],v[tail].y=i;
    }
    for(int i=k; i<=n; i++){
        while(head<=tail && a[i]>=v[tail].x) tail--;
        tail++; v[tail].x=a[i],v[tail].y=i;
        while(head<=tail && i-k+1>v[head].y) head++;
        mx[i-k+1] = v[head].x;
    }
}

int main(){
    cin >> n >> k;
    for(int i=1; i<=n; i++)
        a[i] = read();
    getmin();
    getmax();

    cout << mi[1];
    for(int i=2; i<=(n-k+1); i++)
        cout << " " << mi[i];
    cout << endl;

    cout << mx[1];
    for(int i=2; i<=(n-k+1); i++)
        cout << " " << mx[i];
    cout << endl;

    return 0;
}

 

优先队列的写法

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e6+10;

int a[maxn],mi[maxn],mx[maxn];

struct node1{
    int i;
    bool operator<(const node1& rhs) const{
        return a[i]>a[rhs.i];
    }
};

struct node2{
    int i;
    bool operator<(const node2& rhs) const{
        return a[i]<a[rhs.i];
    }
};

priority_queue<node1> q1;
priority_queue<node2> q2;

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; i++)
        a[i] = read();
    for(int i=1; i<k; i++){
        q1.push(node1{i});
        q2.push(node2{i});
    }

    int cnt1=0,cnt2=0;

    for(int i=k; i<=n; i++){
        q1.push(node1{i});
        q2.push(node2{i});

        while(i-q1.top().i >= k) q1.pop();
        mi[cnt1++] = a[q1.top().i];

        while(i-q2.top().i >= k) q2.pop();
        mx[cnt2++] = a[q2.top().i];
    }

    for(int i=0; i<cnt1; i++){
        cout << mi[i];
        if(i==cnt1-1) cout << endl;
        else cout << " ";
    }

    for(int i=0; i<cnt2; i++){
        cout << mx[i];
        if(i==cnt2-1) cout << endl;
        else cout << " ";
    }

    return 0;
}