单调队列优化dp(四)——#10178. 「一本通 5.5 例 4」旅行问题

题目链接:https://loj.ac/problem/10178
解题思路
先考虑顺时针,定义a[i]=p[i]-d[i],a[i]表示每次行驶改变了多少汽油,sum[i]为其前缀和,如果[1,n]中有sum[i]<0,则不能完成。否则,可以完成。然后一个长度为n的环可以拆分成一段长度为2n-1的链。然后,我们只要保证,每次长度为n的区间的sum最小值大于等于0,就能满足条件。这就可以使用单调队列去做。逆时针,与之类似。
AC代码

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int n;
ll p[N],d[N];
ll a[N<<1],sum[N<<1],b[N<<1],c[N<<1],sum_rev[N<<1];
int flag1[N],flag2[N];
struct node
{
    ll val;
    int pos;
}q[N<<1];
int l,r;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        scanf("%lld%lld",&p[i],&d[i]);
        a[i]=p[i]-d[i];
        //b[n-i+1]=p-d;
       // sum[i]=sum[i-1]+a[i];
    }
    b[1]=p[1]-d[n];
    for(int i=2;i<=n;++i)
    b[i]=p[i]-d[i-1];
    for(int i=1;i<=n;++i)
    c[i]=b[n-i+1];
    for(int i=1;i<n;++i)
    {
        a[i+n]=a[i];
        c[i+n]=c[i];
    }
    for(int i=1;i<(n<<1);++i)
    {
        sum[i]=sum[i-1]+a[i];
        sum_rev[i]=sum_rev[i-1]+c[i];
        //cout<<a[i]<<" "<<sum[i]<<endl;
    }
    for(int i=1;i<=n;++i)
    {
        while(l<r&&sum[i]<=q[r-1].val)
        r--;
        q[r].val=sum[i];
        q[r++].pos=i;
    }
    if(q[l].val>=0)
    flag1[1]=1;
    else
    flag1[1]=0;
    for(int i=n+1;i<(n<<1);++i)
    {
        while(l<r&&(i-q[l].pos+1)>n)
        l++;
        while(l<r&&sum[i]<=q[r-1].val)
        r--;
        q[r].val=sum[i];
        q[r++].pos=i;
        //cout<<q[l].val<<" "<<q[l].pos<<endl;
        //cout<<q[l].val<<" "<<sum[i-n]<<endl;
        if(q[l].val-sum[i-n]>=0)
        flag1[i-n+1]=1;
        else
        flag1[i-n+1]=0;
    }
    l=0,r=0;
    for(int i=1;i<=n;++i)
    {
        while(l<r&&sum_rev[i]<=q[r-1].val)
        r--;
        q[r].val=sum_rev[i];
        q[r++].pos=i;
    }
    if(q[l].val>=0)
    flag2[n]=1;
    else
    flag2[n]=0;
    for(int i=n+1;i<(n<<1);++i)
    {
        while(l<r&&(i-q[l].pos+1)>n)
        l++;
        while(l<r&&sum_rev[i]<=q[r-1].val)
        r--;
        q[r].val=sum_rev[i];
        q[r++].pos=i;
        //cout<<q[l].val<<" "<<q[l].pos<<endl;
        //cout<<q[l].val<<" "<<sum[i-n]<<endl;
        if(q[l].val-sum_rev[i-n]>=0)
        flag2[n-(i-n)]=1;
        else
        flag2[n-(i-n)]=0;
    }
    for(int i=1;i<=n;++i)
    {
        if(flag1[i]|flag2[i])
        puts("TAK");
        else
        puts("NIE");
    }
    return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值