codeforces-div1-284-D

题目连接

Traffic Jams in the Land

首先看题过后很容易联想到线段树

每次询问就相当于对一个区间询问,每次修改就是单点修改.

这样的话,对于这颗线段树,每个节点就需要维护一个值来表示通过这个区间内点所需要的时间.

但是怎么去维护这个解呢?

注意到题目中的ai很小(2到6之间)

于是我们先计算出2,3,4,5,6的最小公倍数,60.

然后对于每个节点,维护一个数组,sum[60],其中sum[t]表示的是以t时间进入这段区间,从这段区间走过需要多少时间.

对于区间的更新,

base = 60

REP(i, base)   tree[u].sum[i] = tree[u*2].sum[i] + tree[u*2+1].sum[(i+tree[u*2].sum[i])%base];

其中u*2为左儿子,u*2+1为右儿子

初始化操作也同更新类似.

下面是我的代码

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <vector>
#include <math.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>
using namespace std;

#define FOR(i, j, k) for(int i=(j);i<=(k);i++)
#define REP(i, n) for(int i=0;i<(n);i++)
#define mst(x, y) memset(x, y, sizeof(x));
#define pii pair<int, int>
#define fr first
#define sc second
#define left myleft
#define right myright
#define ll long long
#define ull unsigned long long
#define seed 1331
#define mod ((int)1e9+7)
#define eps 1e-5
#define pdd pair<double, double>
#define maxn 100008

const int base=60;
int n, m, a[maxn];
struct Node{
    int l, r, sum[base];
}tree[maxn*4];

void init(int u, int x){
    REP(i, base) if(i % x == 0)
        tree[u].sum[i] = 2;
        else tree[u].sum[i] = 1;
}

void update(int u){
    REP(i, base)
    tree[u].sum[i] = tree[u*2].sum[i]
     + tree[u*2+1].sum[(i+tree[u*2].sum[i])%base];
}

void build(int u, int l, int r){
    tree[u].l=l; tree[u].r=r;
    if(l == r){
        init(u, a[l]);
        return ;
    }
    int mid = l+r>>1;
    build(u*2, l, mid);
    build(u*2+1, mid+1, r);
    update(u);
}

void modify(int u, int x, int v){
    if(tree[u].l == tree[u].r){
        init(u, v);
        return ;
    }
    int mid = tree[u].l+tree[u].r>>1;
    if(x <= mid) modify(u*2, x, v);
    else modify(u*2+1, x, v);
    update(u);
}

int query(int u, int l, int r, int flag){
    if(tree[u].l == l && tree[u].r == r) return tree[u].sum[flag];
    int mid = tree[u].l+tree[u].r>>1;
    if(r <= mid) return query(u*2, l, r, flag);
    else if(l > mid) return query(u*2+1, l, r, flag);
    else{
        int t = query(u*2, l, mid, flag);
        int tt = query(u*2+1, mid+1, r, (t+flag)%base);
        return t + tt;
    }
}

int main(){
    cin>>n;
    FOR(i, 1, n)cin>>a[i];
    build(1, 1, n);
    cin>>m;
    while(m --){
        char op[20];
        int x, y;
        cin>>op>>x>>y;
        if(op[0] == 'C') modify(1, x, y);
        else printf("%d\n", query(1, x, y-1, 0));
    }
    return 0;
}