poj 2823 单调队列优化DP

注意初始化,最开始的时候队列里应该放num【1】,如果不放的话,会与队列里的第一个比较,这样就不对了

AC代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int num[1110000];
int q[1110000], pos[1110000], head, tail;
int N, K;

void getmin(){
    head = tail = 0;
    q[0] = num[1];
    pos[0] = 1;
    for( int i = 2; i < K + 1; i++ ){
        while( head <= tail && num[i] < q[tail] ) tail--;
        tail++;
        q[tail] = num[i];
        pos[tail] = i;
    }
    printf( "%d", q[head] );
    for( int i = K + 1; i <= N; i++ ){
        while( head <= tail && num[i] < q[tail] ) tail--;
        tail++;
        q[tail] = num[i];
        pos[tail] = i;
        while( head <= tail && pos[head] < i - K + 1 )  head++;
        printf( " %d", q[head] );
    }
}

void getmax(){
    head = tail = 0;
    q[0] = num[1];
    pos[0] = 1;
    for( int i = 2; i < K + 1; i++ ){
        while( head <= tail && num[i] > q[tail] ) tail--;
        tail++;
        q[tail] = num[i];
        pos[tail] = i;
    }
    printf( "%d", q[head] );
    for( int i = K + 1; i <= N; i++ ){
        while( head <= tail && num[i] > q[tail] ) tail--;
        tail++;
        q[tail] = num[i];
        pos[tail] = i;
        while( head <= tail && pos[head] < i - K + 1 )  head++;
        printf( " %d", q[head] );
    }
}

int main(){
    while( scanf( "%d%d", &N, &K ) != EOF ){
        for( int i = 1; i <= N; i++ ){
            scanf( "%d", &num[i] );
        }
        if( K > N ){
            K = N;
        }
        getmin();
        cout << endl;
        getmax();
        cout << endl;
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值