继单调队列优化背包问题后有一道比较有趣的单调队列题目,这道题更灵活地考察了单调队列的思维与方法,并且要注意细节!
题目:旅行问题
做法:(破环成链+前缀和+单调队列)*2
因为要考虑顺时针和逆时针,所以要做两遍,并且细节都会有不同。
重点是数组 s [ i ] s[i] s[i] ,经过“油量减去路程”和前缀和的处理后,题意转化为:在 s [ i ] s[i] s[i] 到 s [ i + n − 1 ] s[i+n-1] s[i+n−1] ( 1 ⩽ i ⩽ n ) (1\leqslant i\leqslant n) (1⩽i⩽n) 中不能有小于0的值。这样就变成了单调队列经典的滑动窗口求最值的应用。
顺时针:顺时针要从链的尾部开始更新,然后队列中的最值做前缀和计算时的被减数,因此求的是“滑动窗口”中的最小值,对于每一个小于等于 n n n 的 i i i 先更新后再判断 m i n − s [ i ] min - s[i] min−s[i] 是否大于等于0就可以了。
逆时针:逆时针从链首更新,然后队列中的最值做前缀和计算时的减数,因此求的是“滑动窗口”中的最大值,对于每一个大于 n n n 的 i i i 先判断 s [ i ] − m a x s[i] - max s[i]−max 是否大于等于0再更新就可以了。
注意点:
1、顺时针和逆时针的数组
s
s
s 不一样,要分别处理,见代码;
2、顺时针和逆时针对于链不同起点的处理实际上和破环成链的方式有关,依据个人习惯而定;
3、判断和更新的先后顺序可以这样想:每个节点上的油的值都没有了,只有编号,而每一段路上的代价是油减去路程,所以从后向前更新时要多算一个点,而从前向后时不需要。
代码及注释:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=2000010;
ll s[N];
int o[N],d[N],n,q[N];
bool ans[N];
int main(){
memset(ans,false,sizeof(ans));
cin>>n;
for(int i=1;i<=n;i++) scanf("%d%d",&o[i],&d[i]);
int hh=0,tt=-1;
//clockwise
for(int i=1;i<=n;i++) s[i]=s[i+n]=o[i]-d[i]; //ok
for(int i=1;i<=2*n;i++) s[i]+=s[i-1];
for(int i=2*n;i>=1;i--){ //倒序做的原因:因为顺时针做时要求知道i到i+n的范围内的最小值
if(hh<=tt && q[hh]>=i+n) hh++; //*
while(hh<=tt && s[q[tt]]>=s[i]) tt--;
q[++tt]=i;
if(i<=n){
if(s[i-1]<=s[q[hh]]) ans[i]=true;
}
}
hh=0,tt=-1;
//anticlockwise
d[0]=d[n]; //*
for(int i=1;i<=n;i++) s[i]=s[i+n]=o[i]-d[i-1];
for(int i=1;i<=2*n;i++) s[i]+=s[i-1];
for(int i=1;i<=2*n;i++){
if(hh<=tt && q[hh]<=i-n) hh++; //*
if(i>n){
if(s[i]-s[q[hh]]>=0) ans[i-n]=true;
}
while(hh<=tt && s[q[tt]]<=s[i]) tt--; //逆时针要求最大值是因为s[i]做被减数
q[++tt]=i;
}
for(int i=1;i<=n;i++)
if(ans[i]) printf("TAK\n");
else printf("NIE\n");
return 0;
}
细节!!!