BZOJ5254 FJWC2018红绿灯（线段树）

　　注意到一旦在某个路口被红灯逼停，剩下要走的时间是固定的。容易想到预处理出在每个路口被逼停后到达终点的最短时间，这样对于每个询问求出其最早在哪个路口停下就可以了。对于预处理，从下一个要停的路口倒推即可。问题只剩下如何求出下一个要停的路口，这相当于求满足d_i,j%(g+r)>=g的最小j，对d做一个前缀和，那么显然满足条件的是一个值域区间，线段树维护值域区间的路口最小编号即可。对于询问要处理的也与此类似。注意家里没有红灯。（？）

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
#define N 50010
char getc(){char c=getchar();while ((c<'A'||c>'Z')&&(c<'a'||c>'z')&&(c<''||c>'')) c=getchar();return c;}
int gcd(int n,int m){return m==?n:gcd(m,n%m);}
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
int n,m,g,r,a[N],root,cnt;
ll d[N],s[N];
struct data{int l,r,x;
}tree[N<<];
void ins(int &k,int l,int r,int p,int i)
{
    if (!k) k=++cnt,tree[k].x=n+;
    tree[k].x=min(tree[k].x,i);
    if (l==r) return;
    int mid=l+r>>;
    if (p<=mid) ins(tree[k].l,l,mid,p,i);
    else ins(tree[k].r,mid+,r,p,i);
}
int query(int k,int l,int r,int x,int y)
{
    if (!k) return n+;
    if (l==x&&r==y) return tree[k].x;
    int mid=l+r>>;
    if (y<=mid) return query(tree[k].l,l,mid,x,y);
    else if (x>mid) return query(tree[k].r,mid+,r,x,y);
    else return min(query(tree[k].l,l,mid,x,mid),query(tree[k].r,mid+,r,mid+,y));
}
int getnxt(ll x)
{
    x%=g+r;if (x<) x+=g+r;
    if (x<=g) return query(root,,g+r-,g-x,g+r--x);
    else return min(query(root,,g+r-,,g+r--x),query(root,,g+r-,g+r-(x-g),g+r-));
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj5254.in","r",stdin);
    freopen("bzoj5254.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read(),g=read(),r=read();
    for (int i=;i<=n;i++) s[i+]=s[i]+(a[i]=read());
    for (int i=n;i>=;i--)
    {
        int nxt=getnxt(-s[i]);
        if (nxt==n+) d[i]=s[n+]-s[i];
        else d[i]=d[nxt]+((s[nxt]-s[i])/(g+r)+)*(g+r);
        if (i) ins(root,,g+r-,s[i]%(g+r),i);
    }
    m=read();
    while (m--)
    {
        int x=read();
        int nxt=getnxt(x);
        if (nxt==n+) printf(LL,x+s[n+]);
        else printf(LL,d[nxt]+((s[nxt]+x)/(g+r)+)*(g+r));
    }
    return ;
}