bzoj3211: 花神遊歷各國(線段樹)
阿新 • • 發佈:2018-11-19
輸出格式:每次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; }