题目链接:https://loj.ac/problem/10175
解题思路
单调队列优化dp模板题,我们发现因为长度是固定的,每次只是少了前面一个,多了后面一个,这满足单调队列的性质,每次查询队首即可。
AC代码
#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1e6+5;
int n,k;
int a[maxn];
struct node
{
int pos,val;
}q1[maxn],q2[maxn];
int l1,r1,l2,r2;
int ans1[maxn],ans2[maxn];
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
for(int i=1;i<=k;++i)
{
while(l1<r1&&q1[r1-1].val>=a[i])
r1--;
q1[r1].val=a[i];
q1[r1++].pos=i;
while(l2<r2&&q2[r2-1].val<=a[i])
r2--;
q2[r2].val=a[i];
q2[r2++].pos=i;
}
ans1[1]=q1[l1].val;
ans2[1]=q2[l2].val;
for(int i=k+1;i<=n;++i)
{
while(l1<r1&&(i-q1[l1].pos+1)>k)
{
l1++;
//cout<<q1[l1].pos<<" "<<q1[l1].val<<endl;
}
while(l1<r1&&q1[r1-1].val>=a[i])
r1--;
q1[r1].val=a[i];
q1[r1++].pos=i;
ans1[i-k+1]=q1[l1].val;
while(l2<r2&&(i-q2[l2].pos+1)>k)
{
//cout<<i<<" "<<q2[l2].pos<<" "<<q2[l2].val<<endl;
l2++;
}
while(l2<r2&&q2[r2-1].val<=a[i])
r2--;
q2[r2].val=a[i];
q2[r2++].pos=i;
ans2[i-k+1]=q2[l2].val;
}
for(int i=1;i<=n-k+1;++i)
printf("%d ",ans1[i]);
puts("");
for(int i=1;i<=n-k+1;++i)
printf("%d ",ans2[i]);
puts("");
return 0;
}