题目链接: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;
}