1. 程式人生 > >bzoj3211: 花神遊歷各國(線段樹)

bzoj3211: 花神遊歷各國(線段樹)

bzoj3211

輸出格式:每次x=1時,每行一個整數,表示這次旅行的開心度

解析:已經寫了好幾次這種題了,這次來水水部落格 QAQ~
   不難發現對於數列中的每一個數,被進行操作後改變的次數是很少的。
   於是可以記錄下每一段區間是否已經全部變成了1(變成1後再進行運算元仍不變),對於全是1的區間不用再進行操作。
   這樣我們可以對每一個不全是1的區間暴力修改,最後統計一下和即可。

程式碼如下:

#include<cstdio>
#include<algorithm>
#include<cmath>
#define ll long long
#define lc o << 1
#define rc o << 1 | 1
using namespace std;
 
const int maxn = 1e5 + 5;
int n, a[maxn], q, bj[maxn * 4];
ll sum[maxn * 4];
 
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 maintain(int o) { //維護區間 
    sum[o] = sum[lc] + sum[rc];
    if (bj[lc] && bj[rc]) bj[o] = 1; //左右都是1,那麼該區間也是1 
}
 
void build(int o, int l, int r) { //建樹 
    if (l == r) {
      sum[o] = a[l];
      if (sum[o] == 1 || sum[o] == 0) bj[o] = 1; //是0或1就將標記變為1 
      return;
    }
    int mid = l + r >> 1;
    build(lc, l, mid); build(rc, mid + 1, r);
    maintain(o);
}
 
void modify(int o, int l, int r, int ql, int qr) { //區間修改 
    if (bj[o]) return;
    if (l == r) {
      sum[o] = sqrt(sum[o]);
      if (sum[o] == 1) bj[o] = 1; //變為了1就將標記改為1 
      return;
    }
    int mid = l + r >> 1;
    if (ql <= mid) modify(lc, l, mid, ql, qr);
    if (qr > mid) modify(rc, mid + 1, r, ql, qr);
    maintain(o); 
}
 
ll query(int o, int l, int r, int ql, int qr) { //區間查詢 
    if (ql <= l && qr >= r) return sum[o];
    int mid = l + r >> 1; ll ans = 0;
    if (ql <= mid) ans += query(lc, l, mid, ql, qr);
    if (qr > mid) ans += query(rc, mid + 1, r, ql, qr);
    return ans;
} 
 
int main() {
    n = read();
      for (int i = 1; i <= n; ++ i) a[i] = read();
    build(1, 1, n);
    q = read();
      while (q --) {
        int opt = read(), x = read(), y = read();
          if (opt == 1) {
              printf("%lld\n", query(1, 1, n, x, y));
            }
          else {
            modify(1, 1, n, x, y);
          }
      }
    return 0;
}