CodeForces 498 D.Traffic Jams in the Land（线段树）

Description

$n+1$座城市，$i$城市和$i+1$城市之间有一条权值为$a_{i}$的道路，设到达$i$城市的时间是$t$，如果$a_{i}|t$则通过该道路到$i+1$城市用时为2，否则用时为1，有两种操作

$A\ x\ y:$查询$0$时刻从$x$城市出发到$y$城市的用时

$C\ x\ y:$将$x$城市到$x+1$城市道路的权值变为$y$

Input

第一行一个整数$n$表示道路数，然后输入$n$个整数$a_{i}$表示第$i$条道路的权值，然后输入一整数$q$表示操作数，之后$q$行每行一个操作$(1\leq n\leq 10^{5},2\leq a_{i}\leq 6,1\leq q\leq 10^{5})$

Output

对于每次查询操作，输出一个答案

Sample Input

10
2 5 3 2 3 5 3 4 2 4
10
C 10 6
A 2 6
A 1 3
C 3 4
A 3 11
A 4 9
A 5 6
C 7 3
A 8 10
A 2 5

Sample Output

5
3
14
6
2
4
4

Solution

由于$a_{i}$的范围是$[2,6]$，故到达一个路口的时间相差$60$对通过该道路的用时没有影响$(60=lcm(2,3,4,5,6))$

用线段树维护区间和，每个区间节点维护$60$个值，$sum[t][i]$表示到达$t$节点所代表区间左端点的时间为$i$(模$60$)时通过该区间所用时间，区间合并和区间查询时，到右子树的时间为到左子树的时间加上经过左子树的时间

Code

#include<cstdio>
using namespace std;
#define maxn 100005
#define ls (t<<1)
#define rs ((t<<1)|1)
int n,q,a[maxn];
int Sum[maxn<<2][60];
void push_up(int t)
{
    for(int i=0;i<60;i++)
        Sum[t][i]=Sum[ls][i]+Sum[rs][(i+Sum[ls][i])%60];
}
void build(int l,int r,int t)
{
    if(l==r)
    {
        for(int i=0;i<60;i++)
            if(i%a[l])Sum[t][i]=1;
            else Sum[t][i]=2;
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,ls),build(mid+1,r,rs);
    push_up(t);
}
void update(int x,int l,int r,int t,int v)
{
    if(l==r)
    {
        for(int i=0;i<60;i++)
            if(i%v)Sum[t][i]=1;
            else Sum[t][i]=2;
        return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)update(x,l,mid,ls,v);
    else update(x,mid+1,r,rs,v);
    push_up(t);
}
int query(int L,int R,int l,int r,int t,int st)
{
    if(L<=l&&r<=R)return Sum[t][st];
    int mid=(l+r)/2;
    if(R<=mid)return query(L,R,l,mid,ls,st);
    else if(L>mid)return query(L,R,mid+1,r,rs,st);
    else
    {
        int temp=query(L,R,l,mid,ls,st);
        return temp+query(L,R,mid+1,r,rs,(st+temp)%60);
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        build(1,n,1);
        scanf("%d",&q);
        while(q--)
        {
            char op[3];
            int x,y;
            scanf("%s%d%d",op,&x,&y);
            if(op[0]=='A')printf("%d\n",query(x,y-1,1,n,1,0));
            else update(x,1,n,1,y);
        }
    }
    return 0;
}