bzoj1798

題目描述：給定n個數的序列，有三種操作。
?????1、將一段區間乘上c
?????2、將一段區間加上c
?????3、求一段區間的和

輸入格式：第一行兩個整數第一行兩個整數N和P(1≤P≤1000000000）。第二行含有N個非負整數,從左到右依次為a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。第三行有一個整數M，表示操作總數。從第四行開始每行描述一個操作，輸入的操作有以下三種形式： 操作1：“1 t g c”(不含雙引號)。表示把所有滿足t≤i≤g的ai改為ai×c(1≤t≤g≤N,0≤c≤1000000000)。 操作2：“2 t g c”(不含雙引號)。表示把所有滿足t≤i≤g的ai改為ai+c (1≤t≤g≤N,0≤c≤1000000000)。 操作3：“3 t g”(不含雙引號)。詢問所有滿足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相鄰兩數之間用一個空格隔開，每行開頭和末尾沒有多余空格。

輸出格式：對每個操作3，按照它在輸入中出現的順序，依次輸出一行一個整數表示詢問結果。

輸入樣例：
7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7

輸出樣例：
2
35
8

解析：線段樹的模板題。關鍵在於如何處理線段樹的乘和加。
???可以將每一個數變為\(a \times b + c\) 的形式，在每一次乘的修改時，直接將和乘c，並將加和乘的標記都乘c即可。而加的修改則只需要將加的標記加上c即可。
???在標記下傳時，將該點子樹的乘標記乘上該點的乘標記；將子樹的加標記先乘上該點的乘標記，再加上該點的加標記即可。

代碼如下：

#include<cstdio>
#define lc o << 1
#define rc o << 1 | 1
using namespace std;

const int maxn = 1e5 + 5;
int n, m, p, a[maxn], sum[maxn << 2], bj1[maxn << 2], bj2[maxn << 2]; //bj1:乘標記，bj2:加標記 

int read(void) {
    char c; while (c = getchar(), c < '0' || c >'9'); int x = c - '0';
    while (c = getchar(), c >= '0' && c <= '9') x = x * 10 + c - '0'; return x;
}

void build(int o, int l, int r) { //建樹 
    bj1[o] = 1; //乘標記的初始值為1 
    if (l == r) {
      sum[o] = a[l] % p; 
      return;
    }
    int mid = l + r >> 1;
    build(lc, l, mid); build(rc, mid + 1, r);
    sum[o] = (sum[lc] + sum[rc]) % p;
}

void pushdown(int o, int l, int r) { //標記下傳 
    int mid = l + r >> 1;
    sum[lc] = 1ll * sum[lc] * bj1[o] % p; sum[lc] += 1ll * bj2[o] * (mid - l + 1) % p; sum[lc] %= p;
    sum[rc] = 1ll * sum[rc] * bj1[o] % p; sum[rc] += 1ll * bj2[o] * (r - mid) % p; sum[rc] %= p;
    bj1[lc] = 1ll * bj1[lc] * bj1[o] % p;
    bj1[rc] = 1ll * bj1[rc] * bj1[o] % p;
    bj2[lc] = 1ll * bj2[lc] * bj1[o] % p; bj2[lc] = (bj2[lc] + bj2[o]) % p;
    bj2[rc] = 1ll * bj2[rc] * bj1[o] % p; bj2[rc] = (bj2[rc] + bj2[o]) % p;
    bj1[o] = 1; bj2[o] = 0;
} 

void modify1(int o, int l, int r, int ql, int qr,int c) {
    if (ql <= l && qr >= r) { //兩個標記都要乘上c 
      sum[o] = 1ll * sum[o] * c % p; 
      bj1[o] = 1ll * bj1[o] * c % p;
      bj2[o] = 1ll * bj2[o] * c % p;
      return;
    }
    int mid = l + r >> 1;
    if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
    if (ql <= mid) modify1(lc, l, mid, ql, qr, c);
    if (qr > mid) modify1(rc, mid + 1, r, ql, qr, c);
    sum[o] = (sum[lc] + sum[rc]) % p;
}

void modify2(int o, int l, int r, int ql, int qr, int c) {
    if (ql <= l && qr >= r) { //加標記直接加上c 
      sum[o] += 1ll * c * (r - l + 1) % p; sum[o] %= p;
      bj2[o] = (bj2[o] + c) % p;
      return;
    }
    int mid = l + r >> 1;
    if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
    if (ql <= mid) modify2(lc, l, mid, ql, qr, c);
    if (qr > mid) modify2(rc, mid + 1, r, ql, qr, c);
    sum[o] = (sum[lc] + sum[rc]) % p;
}

int query(int o, int l, int r, int ql, int qr) {
    if (ql <= l && qr >= r) return sum[o];
    int mid = l + r >> 1, ans = 0;
    if (bj1[o] != 1 || bj2[o]) pushdown(o, l, r);
    if (ql <= mid) ans += query(lc, l, mid, ql, qr); ans %= p;
    if (qr > mid) ans += query(rc, mid + 1, r, ql, qr); ans %= p;
    return ans;
}

int main() {
    n = read(); p = read();
      for (int i = 1; i <= n; ++ i) a[i] = read();
    build(1, 1, n);
    m = read();
      while (m --) {
        int opt = read();
          if (opt == 1) {
              int x = read(), y = read(), c = read();
              modify1(1, 1, n, x, y, c);
            }
          else if (opt == 2) {
            int x = read(), y = read(), c = read();
            modify2(1, 1, n, x, y, c);
          }
          else {
            int x = read(), y = read();
            printf("%d\n", query(1, 1, n, x, y));
          }
      }
    return 0;
}
 

bzoj1798: [Ahoi2009]Seq 維護序列seq（線段樹）