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

Description

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

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

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

Input

第一行一个整数n表示道路数，然后输入n个整数ai表示第i条道路的权值，然后输入一整数q表示操作数，之后q行每行一个操作(1≤n≤105,2≤ai≤6,1≤q≤105)

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

由于ai的范围是[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;}