[Bzoj5254][Fjwc2018]红绿灯（线段树）

5254: [Fjwc2018]红绿灯

---

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 31  Solved: 24
[Submit][Status][Discuss]

Description

---

泰迪每天都要通过一条路从家到学校，这条路的起点是泰迪家，终点则是学校。

这条路中间还有n个路口，从第i-1个路口走到第i个路口需要di秒，每个路口都有一个红绿灯。更具体的，绿灯持

续时间是g秒，红灯持续时间是r秒。每天从第0秒开始，所有灯都是绿灯，持续g秒之后变为红灯，再过r秒变成绿

灯，以此类推，并且同一时刻所有灯都是相同状态。当泰迪到达一个路口，若是绿灯则可直接通过，若是红灯则需

原地等待至绿灯。若到达某一路口时灯的状态正好发生改变，则视达到路口时灯的颜色为其改变后的颜色，例如第

g秒到达一个路口则视为遇到红灯。

现在泰迪预计了接下来q天从家出发的时间，第j天将会在第tj秒从家出发，他希望你告诉他每天到达学校的最早时

间。你可以假定一天内泰迪一定可以到达学校。

Input

---

第一行三个正整数n,g,r表示路口数以及绿灯、红灯持续的时间。

第二行n+1个正整数di表示相邻路口间的通行时间

d1表示从起点到第一个路口所需的时间，dn+1表示第n个路口到终点的所需时间。

第三行一个正整数q表示询问天数。

第四行q个非负整数tj表示每天的出发时间

n, q ≤ 5 × 10^4 , 2 ≤ g, r ≤ 10^9 , di, tj ≤ 10^9

Output

---

共q行，对于每次询问输出一行一个整数表示答案。

Sample Input

---

Sample Output

---

HINT

---

 

Source

---

By ExfJoe

分析：

---

先不考虑红绿灯，处理出到每个路口前缀和，离散化后建权值线段树，倒着来扫一遍，找到每个点出发第一个红灯在哪。

预处理出以每个点从时刻0出发，考虑红绿灯时到学校的路径（即f数组）

然后每次询问只用找到第一个红灯出现的地方，和预处理相同。

AC代码：

---

# include <iostream>
# include <cstdio>
# include <algorithm>
using namespace std;
typedef long long LL;
const int N = 5e4 + ;
int block[N],tot,n,m,Q;LL G,C,P,d[N],b[N],s[N],f[N],w;
int mi[N << ];
void push(int x){mi[x] = min(mi[x << ],mi[x <<  | ]);}
void updata(int L,int l,int r,int rt,int v)
{
    if(l == r){mi[rt] = v;return;}
    int mid = l + r >> ;
    if(L <= mid)updata(L,l,mid,rt << ,v);
    else updata(L,mid + ,r,rt <<  | ,v);
    push(rt);
}
int Query(int L,int R,int l,int r,int rt)
{
    if(L > R)return n + ;
    if(L <= l && r <= R)return mi[rt];
    int mid = l + r >> ;
    if(L > mid)return Query(L,R,mid + ,r,rt <<  | );
    if(R <= mid)return Query(L,R,l,mid,rt << );
    return min(Query(L,R,l,mid,rt << ),Query(L,R,mid + ,r,rt <<  | ));
}
int query(LL w)
{
    w = (w % P + P) % P;
    LL L,R;int lc,rc,k;
    L = (G + w) % P,R = (P -  + w) % P;
    lc = lower_bound(b + ,b + m + ,L) - b;
    rc = upper_bound(b + ,b + m + ,R) - b - ;
    if(L <= R)k = Query(lc,rc,,m,);
    else k = min(Query(lc,m,,m,),Query(,rc,,m,));
    return k;
}
void add(int x,int y,LL bac)
{
    f[x] = f[y] + s[y] - s[x];
    if(y != n + )f[x] = f[x] + P - (s[y] - s[x] + bac) % P;
}
int main()
{
    scanf("%d %lld %lld",&n,&G,&C);P = G + C;
    for(int i = ;i <= n + ;i++)scanf("%lld",&d[i]);
    for(int i = ;i <= n + ;i++)s[i] = s[i - ] + d[i],b[i] = s[i] % P;
    b[n + ] = ;b[n + ] = P;
    sort(b + ,b + n + );
    m = unique(b + ,b + n + ) - b - ;
    for(int i = ;i <= (m << );i++)mi[i] = n + ;
    int k;
    for(int i = n;i >= ;i--)
    {
        k = query(s[i] % P);
        add(i,k,);
        updata(lower_bound(b + ,b + m + ,s[i] % P) - b,,m,,i);
    }
    scanf("%d",&Q);
    while(Q--)
    {
      scanf("%lld",&w);
      k = query(-w);
      add(,k,w);
      printf("%lld\n",f[] + w);
    }
}