Lightoj 1411 - Rip Van Winkle`s Code(線斷樹)

題意

就是用快速的方式實現在面的四種操作。

long long data[250001];
void A( int st, int nd ) {
    for( int i = st; i <= nd; i++ ) data[i] = data[i] + (i - st + 1);
}
void B( int st, int nd ) {
    for( int i = st; i <= nd; i++ ) data[i] = data[i] + (nd - i + 1);
}
void C( int st, int nd, int x ) {
    for( int i = st; i <= nd; i++ ) data[i] = x;
}
long long S( int st, int nd ) {
    long long res = 0;
    for( int i = st; i <= nd; i++ ) res += data[i];
    return res;
}

思路

A.可以看成是區間每個數+(1-st)，然後都加上自己的下標一次。
B.可以看成是區間每個數+(1+nd)，然後都減去自己的下標一次。
C.區間覆蓋。
D.求和。
A、B的操作分成了兩個操作，所以就變得普通了，對於下標就是等差數列求和問題了。
但是向下更新的時候應該先考慮覆蓋，再考慮加減。

const int INF = 0x3f3f3f3f;
int nCase = 0;
const int maxn = 250000 + 123;
struct SegmentTreeNode {
    int l, r;
    LL add1, add2, add3;
    LL sum;
}p[maxn<<2];
inline void pushup(int rt) {
    p[rt].sum = p[lson].sum + p[rson].sum;
}
inline void pushdown(int rt) {
    //優先考慮覆蓋操作
    if (p[rt].add1 != -INF) {
        p[lson].add2 = p[lson].add3 = p[rson].add2 = p[rson].add3 = 0;
        p[lson].add1 = p[rson].add1 = p[rt].add1;
        p[lson].sum = p[rt].add1 * (p[lson].r - p[lson].l + 1);
        p[rson].sum = p[rt].add1 * (p[rson].r - p[rson].l + 1);
        p[rt].add1 = -INF;
    }
    if (p[rt].add2) {
        //每個數加減
        p[lson].add2 += p[rt].add2;
        p[rson].add2 += p[rt].add2;
        p[lson].sum += p[rt].add2 * (p[lson].r - p[lson].l + 1);
        p[rson].sum += p[rt].add2 * (p[rson].r - p[rson].l + 1);
        p[rt].add2 = 0;
    }
    if (p[rt].add3) {
        //該區間下標加減次數,等差數列求和
        p[lson].add3 += p[rt].add3;
        p[rson].add3 += p[rt].add3;
        p[lson].sum += p[rt].add3 * (p[lson].r - p[lson].l + 1) * (p[lson].l + p[lson].r) / 2;
        p[rson].sum += p[rt].add3 * (p[rson].r - p[rson].l + 1) * (p[rson].l + p[rson].r) / 2;
        p[rt].add3 = 0;
    }
}
void build(int rt,int L, int R) {
    p[rt] = SegmentTreeNode{L, R, -INF, 0, 0, 0};
    if (L == R) return ;
    int mid = (L + R) >> 1;
    build(lson, L, mid), build(rson, mid + 1, R);
}
void updata(int rt,int L, int R, int id, int val) {
    if (L <= p[rt].l && p[rt].r <= R) {
        if (id == 1) {
            p[rt].add1 = (LL) val;
            p[rt].add2 = p[rt].add3 = 0;
            p[rt].sum = (LL)val * (p[rt].r - p[rt].l + 1);
        }else if (id == 2) {
            p[rt].add2 += (LL) val;
            p[rt].sum += (LL)val * (p[rt].r - p[rt].l + 1);
        }else {
            p[rt].add3 += val;
            p[rt].sum += (LL)val * (p[rt].r - p[rt].l + 1) * (p[rt].l + p[rt].r) / 2;
        }
        return ;
    }
    pushdown(rt);
    int mid = (p[rt].l + p[rt].r) >> 1;
    if (L <= mid) updata(lson, L, R, id, val);
    if (R > mid) updata(rson, L, R, id, val);
    pushup(rt);
    // printf("[區間[%d, %d], sum = %lld]\n", p[rt].l, p[rt].r, p[rt].sum);
}
LL Query(int rt,int L, int R) {
    if (L <= p[rt].l && p[rt].r <= R) return p[rt].sum;
    pushdown(rt);
    LL res = 0LL;
    int mid = (p[rt].l + p[rt].r) >> 1;
    if (L <= mid) res += Query(lson, L, R);
    if (R > mid) res += Query(rson, L, R);
    pushup(rt);
    return res;
}
int main(int argc, const char * argv[])
{
    int kase;cin >> kase;
    while(kase--) {
        build(1, 1, maxn - 1);
        int n;scanf("%d", &n);
        printf("Case %d:\n", ++nCase);
        while(n--) {
            char op[3];
            scanf("%s", op);
            if (op[0] == 'A') {
                int l, r;scanf("%d%d", &l, &r);
                updata(1, l, r, 2, 1 - l);
                updata(1, l, r, 3, 1);
            }else if (op[0] == 'B') {
                int l, r;scanf("%d%d", &l, &r);
                updata(1, l, r, 2, 1 + r);
                updata(1, l, r, 3, -1);
            }else if (op[0] == 'C') {
                int l, r, val;
                scanf("%d%d%d", &l, &r, &val);
                updata(1, l, r, 1, val);
            }else {
                int l, r;scanf("%d%d", &l, &r);
                printf("%lld\n", Query(1, l, r));
            }
        }
    }
    return 0;
}