HDU 4267 A Simple Problemwith Integers(セグメントツリー)


タイトル:
N<=5×104,区間[a,b]内で(i−a)%k=0(i∈[a,b])を満たす数Aiを毎回c増加させ,最後の単点クエリ値
分析:
  • シフトは、i%k=a%kを更新するたびに、
  • を個別に維持することを発見した.
  • 裸の区間増減、単点照会のモデルですが、BITを久しぶりに書いていないので見えず、線分ツリーを考えて行きましたが、単点照会しかなく、区間増減値の和を維持しました.
  • という問題は意図的に空間を押さえている−−MLEは多くの髪を送り、最後にメンテナンス区間の増減値さえすればよいと反応したが、線分樹かMLEかでadd[k][mod]をadd[55]に離散化するしかなかった.
  • ブルーBIT少定数倍空間は簡単に過ぎ、離散化
  • を必要としない.
    コード:セグメントツリー:
    //
    // Created by TaoSama on 2015-09-30
    // Copyright (c) 2015 TaoSama. All rights reserved.
    //
    //#pragma comment(linker, "/STACK:1024000000,1024000000")
    #include <algorithm>
    #include <cctype>
    #include <cmath>
    #include <cstdio>
    #include <cstdlib>
    #include <cstring>
    #include <iomanip>
    #include <iostream>
    #include <map>
    #include <queue>
    #include <string>
    #include <set>
    #include <vector>
    
    using namespace std;
    #define pr(x) cout << #x << " = " << x << " "
    #define prln(x) cout << #x << " = " << x << endl
    const int N = 5e4 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;
    
    int n, q, A[N];
    int id[11][10];
    
    #define root 1, n, 1
    #define lson l, m, rt << 1
    #define rson m + 1, r, rt << 1 | 1
    
    int add[N << 2][55];
    bool lazy[N << 2];
    
    int cur;
    
    void push_down(int rt) {
        if(lazy[rt]) {
            for(int i = 0; i < 55; ++i) {
                add[rt << 1][i] += add[rt][i];
                add[rt << 1 | 1][i] += add[rt][i];
                add[rt][i] = 0;
            }
            lazy[rt << 1] = lazy[rt << 1 | 1] = lazy[rt];
            lazy[rt] = 0;
        }
    }
    void build(int l, int r, int rt) {
        lazy[rt] = 0;
        for(int i = 0; i < 55; ++i) add[rt][i] = 0;
        if(l == r) return;
        int m = l + r >> 1;
        build(lson);
        build(rson);
    }
    void update(int L, int R, int v, int l, int r, int rt) {
        if(L <= l && r <= R) {
            lazy[rt] = 1;
            add[rt][cur] += v;
            return;
        }
        int m = l + r >> 1;
        push_down(rt);
        if(L <= m) update(L, R, v, lson);
        if(R > m) update(L, R, v, rson);
    }
    int query(int o, int l, int r, int rt) {
        if(l == r) {
            int ret = A[l];
            for(int k = 1; k <= 10; ++k) ret += add[rt][id[k][o % k]];
            return ret;
        }
        int m = l + r >> 1;
        push_down(rt);
        if(o <= m) return query(o, lson);
        else return query(o, rson);
    }
    
    
    void gao() {
        int cnt = 0;
        for(int i = 1; i <= 10; ++i) for(int j = 0; j < i; ++j) id[i][j] = cnt++;
        //cnt = 55;
    }
    
    int main() {
    #ifdef LOCAL
        freopen("in.txt", "r", stdin);
    // freopen("out.txt","w",stdout);
    #endif
        ios_base::sync_with_stdio(0);
    
        gao();
        while(scanf("%d", &n) == 1) {
            for(int i = 1; i <= n; ++i) scanf("%d", A + i);
            scanf("%d", &q);
            build(root);
            while(q--) {
                int op, a; scanf("%d%d", &op, &a);
                if(op == 1) {
                    int b, k, c; scanf("%d%d%d", &b, &k, &c);
                    cur = id[k][a % k];
                    update(a, b, c, root);
                } else printf("%d
    "
    , query(a, root)); } } return 0; }

    BIT:
    //
    // Created by TaoSama on 2015-09-30
    // Copyright (c) 2015 TaoSama. All rights reserved.
    //
    //#pragma comment(linker, "/STACK:1024000000,1024000000")
    #include <algorithm>
    #include <cctype>
    #include <cmath>
    #include <cstdio>
    #include <cstdlib>
    #include <cstring>
    #include <iomanip>
    #include <iostream>
    #include <map>
    #include <queue>
    #include <string>
    #include <set>
    #include <vector>
    
    using namespace std;
    #define pr(x) cout << #x << " = " << x << " "
    #define prln(x) cout << #x << " = " << x << endl
    const int N = 5e4 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;
    
    int n, q, A[N];
    int b[11][10][N];
    
    void add(int k, int mod, int i, int v) {
        for(; i <= n; i += i & -i) b[k][mod][i] += v;
    }
    
    int sum(int k, int mod, int i) {
        int ret = 0;
        for(; i; i -= i & -i) ret += b[k][mod][i];
        return ret;
    }
    
    int main() {
    #ifdef LOCAL
        freopen("in.txt", "r", stdin);
    // freopen("out.txt","w",stdout);
    #endif
        ios_base::sync_with_stdio(0);
    
        while(scanf("%d", &n) == 1) {
            for(int i = 1; i <= n; ++i) scanf("%d", A + i);
            scanf("%d", &q);
            memset(b, 0, sizeof b);
            while(q--) {
                int op, a; scanf("%d%d", &op, &a);
                if(op == 1) {
                    int b, k, c; scanf("%d%d%d", &b, &k, &c);
                    add(k, a % k, a, c);
                    add(k, a % k, b + 1, -c);
                } else {
                    int ans = A[a];
                    for(int k = 1; k <= 10; ++k) ans += sum(k, a % k, a);
                    printf("%d
    "
    , ans); } } } return 0; }