[日常训练] 魔法传输

【问题描述】

虽然已经知道了至少要多少魔法值，但是小Y现在最想知道的是如何获得魔法值。
小Y和一群向往魔法的孩子（当然这些孩子们都是不会魔法的）来到了哈利波特的家，大家坐成一排。哈利波特会不时的给大家传输魔法。
哈利每次会选择一个区间，给这个区间里的孩子们传输魔法：最左边的孩子给一点，第二个给两点……哈利有时会突然问你某一个孩子已经有了多少魔法。如果你回答出来，小Y说不定会请你吃汤圆哦！

【输入格式】

第一行两个正整数N,M，表示有N个孩子，哈利有M次操作。
接下来M行，每行代表一个操作。第一个字符为ci，若ci=‘C’则此次操作为送魔法操作，接下来会有两个整数Li,Ri,表示此次送分的区间。若ci=‘Q’则此次操作为询问操作，接下来一个整数xi，表示询问第xi个孩子的当前魔法值。

【输出格式】

对于每组询问输出一行，仅包含一个整数，表示答案对1,000,000,007取mod的结果。

【输入输出样例】

score.in
3 4
C 1 3
Q 2
C 2 3
Q 2 2

score.out
3

【数据规模】

对于30%的数据N,M≤1,000
对于100%的数据N,M≤100,000

【30~80分】 暴力模拟

【100分】

【解法1】线段树

对于线段树上记录的每一个区间$[l, r]$，我们新增三个域，分别表示位置$l$上增加的魔法点，位置$r$上增加的魔法点，以及位置$i(l < i \le r)$对于位置$i - 1$所增加的魔法点。这样我们在修改区间时，打上标记的必然是完整的区间，同时下传时可以根据上面的三个值来确定下传区间的三个值。

【代码1】

#include <iostream>
#include <cstdio>
#define sL (s << 1)
#define sR (s << 1 | 1)

using namespace std;
typedef long long ll;
const ll Mod = 1000000007ll;
const int N = 4e5 + 5;

ll tl[N], tr[N], val[N];
int n, m; 

inline void AddTag(const int &s, const int &l, const int &r, const ll &vl, const ll &vr, const ll &va)
{
    tl[s] += vl; tr[s] += vr; val[s] += va;
}

inline void Down(const int &s, const int &l, const int &r)
{
    if (!tl[s] && !tr[s]) return ;
    int mid = l + r >> 1;
    AddTag(sL, l, mid, tl[s], tl[s] + (mid - l) * val[s], val[s]); 
    AddTag(sR, mid + 1, r, tr[s] - (r - mid - 1) * val[s], tr[s], val[s]);
    tl[s] = tr[s] = val[s] = 0;
}


inline void Modify(const int &s, const int &l, const int &r, const int &x, const int &y, const ll &vl, const ll &vr)
{
    if (x == l && y == r) return AddTag(s, l, r, vl, vr, 1);
    Down(s, l, r); int mid = l + r >> 1;
    if (y <= mid) Modify(sL, l, mid, x, y, vl, vr);
     else if (x > mid) Modify(sR, mid + 1, r, x, y, vl, vr);
      else 
      {
        Modify(sL, l, mid, x, mid, vl, vl + mid - x);
        Modify(sR, mid + 1, r, mid + 1, y, vr - (y - mid - 1), vr);
      } 
}

inline ll Query(const int &s, const int &l, const int &r, const int &x)
{
    if (l == r) return tl[s];
    Down(s, l, r); int mid = l + r >> 1;
    if (x <= mid) return Query(sL, l, mid, x);
     else return Query(sR, mid + 1, r, x); 
}

inline void put(ll x)
{
    if (x > 9) put(x / 10);
    putchar(x % 10 + 48);
}

int main()
{
    freopen("score.in", "r", stdin);
    freopen("score.out", "w", stdout);
    scanf("%d%d", &n, &m); int u, v;
    while (m--)
    {
        char tp;
        while ((tp = getchar()) != 'C' && tp != 'Q');
        if (tp == 'C')
        {
            scanf("%d%d", &u, &v);
            Modify(1, 1, n, u, v, 1, v - u + 1); 
        }
        else 
        {
            scanf("%d", &u);
            put(Query(1, 1, n, u) % Mod), putchar('\n');
        }
    }
    fclose(stdin); fclose(stdout);
    return 0; 
} 

【解法2】 树状数组(代码短，常数小)

对于修改区间$[l, r]$上的位置$i$，它这次增加的魔法点为$i - l + 1$，我们考虑把$i - l + 1$拆成$i + 1$和$-l$分别考虑，然后用两个树状数组维护两个差分（对于位置$i$的魔法点，转化为求区间$[1, i]$的和，则如果想在区间$[l, r ]$的每个位置都增加$x$，即在位置$l$增加$x$，位置$r + 1$减少$x$）。
对于$i + 1$，我们发现其只与位置$i$有关，所以就可以用一个树状数组记录增加$i + 1$的次数$k$，则它对答案的贡献就为$k \times (i + 1)$；对于$-l$，我们同样用另一个树状数组修改即可。

【代码2】

#include <iostream>
#include <cstdio>

using namespace std;
typedef long long ll;
const ll Mod = 1e9 + 7;
const int N = 1e5 + 5;
ll b[N], c[N]; int n, m;

inline void Modify(int x, const ll &y, ll *a) {for (; x <= n; x += x & (-x)) a[x] += y;}
inline int Query(int x, ll *a) {ll res = 0; for (; x; x -= x & (-x)) res += a[x]; return res;}

inline int get()
{
    char ch; int res;
    while ((ch = getchar()) < '0' || ch > '9');
    res = ch - '0';
    while ((ch = getchar()) >= '0' && ch <= '9') 
     res = (res << 3) + (res << 1) + ch - '0';
    return res;
}

inline void put(ll x)
{
    if (x > 9) put(x / 10);
    putchar(x % 10 + 48);
}

int main()
{
    freopen("score.in", "r", stdin);
    freopen("score.out", "w", stdout);
    n = get(); m = get(); int x, y;
    while (m--)
    {
        char tp;
        while ((tp = getchar()) != 'C' && tp != 'Q');
        if (tp == 'C') 
        {
            x = get(); y = get();
            Modify(x, -x, b); Modify(y + 1, x, b);
            Modify(x, 1, c); Modify(y + 1, -1, c);
        }
        else 
        {
            x = get();
            put((Query(x, b) + Query(x, c) * (x + 1)) % Mod), putchar('\n');
        }   
    }
    fclose(stdin); fclose(stdout);
    return 0;
}