[一本通 5.5 例 4」旅行问题

题目描述:

John 打算驾驶一辆汽车周游一个环形公路。公路上总共有 n 车站,每站都有若干升汽油(有的站可能油量为零),每升油可以让汽车行驶一千米。John 必须从某个车站出发,一直按顺时针(或逆时针)方向走遍所有的车站,并回到起点。在一开始的时候,汽车内油量为零,John 每到一个车站就把该站所有的油都带上(起点站亦是如此),行驶过程中不能出现没有油的情况。

任务:判断以每个车站为起点能否按条件成功周游一周。

输入格式:

第一行是一个整数 n,表示环形公路上的车站数;

接下来 n 行,每行两个整数 pi,di,分别表示表示第 i 号车站的存油量和第 i 号车站到下一站的距离。

输出格式

输出共 n 行,如果从第 i 号车站出发,一直按顺时针(或逆时针)方向行驶,能够成功周游一圈,则在第 i 行输出 TAK,否则输出 NIE

输入样例

5
3 1
1 2
5 2
0 1
5 4

输出样例
TAK
NIE
TAK
NIE
TAK

n<=1e6,long long


思路:
设ai=pi-di,即此站到下一站花费(或增加)的油量
用sum记录前缀和,即从开始到现在总共花费(或增加)的油量
只要每n个联系区间内最小值is负数,那么就肯定无法周游一周
暴力的时复为O(n^2)
然而我们发现这是一个连续且序号单调递升的区间
很容易想到用单调队列维护一下
这样时复就是O(n)了
注意:
正着周游一圈可以,反着周游一圈也可以,需要在判断一次
上代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define rep(i,a,b) for(long long i=a;i<=b;i++)
#define N 2000500
using namespace std;
typedef long long ll;
deque<pair<ll,ll> > Q;
ll n,p[N],d[N],p1[N],d1[N],a[N],sum[N],ans[N];
ll work(ll x){
    if(x==n) return 1;
    return 2*n-x+1;
}
int main()
{
    scanf("%lld",&n);
    rep(i,1,n){
        scanf("%lld%lld",&p[i],&d[i]);
        a[i+n]=a[i]=p[i]-d[i];
        sum[i]=sum[i-1]+a[i];
    }
    rep(i,1+n,n+n) sum[i]=sum[i-1]+a[i];
    rep(i,1,2*n-1){
        while(!Q.empty() && Q.front().first+n<=i) Q.pop_front();
        while(!Q.empty() && Q.back().second>=sum[i]) Q.pop_back();
        Q.push_back(make_pair(i,sum[i]));
        if(i>=n){
            if(Q.front().second-sum[i-n]<0) ans[i-n+1]=0;
            else ans[i-n+1]=1;
        }
    }
    p1[1]=p[1]; d1[1]=d[n]; 
    a[1]=a[1+n]=p1[1]-d1[1];
    sum[1]=a[1];
    rep(i,2,n){
        p1[i]=p[n-i+2]; d1[i]=d[n-i+1];
        a[i]=a[i+n]=p1[i]-d1[i];
        sum[i]=sum[i-1]+a[i];
    }
    rep(i,n+1,n*2) sum[i]=sum[i-1]+a[i];
    rep(i,1,2*n-1){
        while(!Q.empty() && Q.front().first+n<=i) Q.pop_front();
        while(!Q.empty() && Q.back().second>=sum[i]) Q.pop_back();
        Q.push_back(make_pair(i,sum[i]));
        if(i>=n) if(Q.front().second-sum[i-n]>=0) ans[work(i)]=1;
    }
    rep(i,1,n){
        if(ans[i]) puts("TAK");
        else puts("NIE");
    }
    return 0;
}
 
原文地址:https://www.cnblogs.com/handsome-zlk/p/10043593.html