PKU A Simple Problem with Integers 3468


코드 가 인터넷 에서 왔 습 니 다 (원본 이 아 닙 니 다)
 
 

  
  
  
  
  1. /*  
  2. Q , , C ,  
  3. , s, d, C , , , , , s 。 , O(logn) .  
  4. Memory: 6760K  
  5. Time: 1579MS  */
  6.  
  7. #include <iostream>  
  8. using namespace std;  
  9. #define ll long long  
  10. const int MAXN = 100001;  
  11. struct {  
  12.     int l, r;  
  13.     ll s, d;  
  14. } nod[MAXN*3];  
  15. int a[MAXN];  
  16. void creat(int t, int l, int r) {  
  17.     if(l == r) {  
  18.         nod[t].l = nod[t].r = l, nod[t].s = a[l], nod[t].d = 0;  
  19.         return;  
  20.     }  
  21.     int m = (l+r) / 2;  
  22.     nod[t].l = l, nod[t].r = r, nod[t].d = 0;  
  23.     creat(t*2, l, m), creat(t*2+1, m+1, r);  
  24.     nod[t].s = nod[t*2].s + nod[t*2+1].s;  
  25. }  
  26. void inc(int t, int l, int r, int c) {  
  27.     if(l == nod[t].l && r == nod[t].r) {nod[t].d += c; return;}  
  28.     if(r <= nod[t*2].r) inc(t*2, l, r, c);  
  29.     else if(l >= nod[t*2+1].l) inc(t*2+1, l, r, c);  
  30.     else inc(t*2, l, nod[t*2].r, c), inc(t*2+1, nod[t*2+1].l, r, c);  
  31.     nod[t].s = nod[t*2].s + nod[t*2].d * (nod[t*2].r-nod[t*2].l+1) + nod[t*2+1].s + nod[t*2+1].d * (nod[t*2+1].r-nod[t*2+1].l+1);  
  32. }  
  33. ll query(int t, int l, int r) {  
  34.     if(nod[t].l == l && nod[t].r == r) return nod[t].s + nod[t].d * (r-l+1);  
  35.     ll sum;  
  36.     if(r <= nod[t*2].r) sum = query(t*2, l, r);  
  37.     else if(l >= nod[t*2+1].l) sum = query(t*2+1, l, r);  
  38.     else sum = query(t*2, l, nod[t*2].r) + query(t*2+1, nod[t*2+1].l, r);  
  39.     return sum + nod[t].d * (r-l+1);  
  40. }  
  41. int main() {  
  42.     int i, n, q, x1, x2, c;  
  43.     char s[2];  
  44.     while(scanf("%d%d", &n, &q) != EOF) {  
  45.         for(i = 1; i <= n; i++) scanf("%d", &a[i]);  
  46.         creat(1, 1, n);  
  47.         while(q--) {  
  48.             scanf("%s%d%d", s, &x1, &x2);  
  49.             if(s[0] == 'C') {scanf("%d", &c); inc(1, x1, x2, c);}  
  50.             else printf("%lld
    "
    , query(1, x1, x2));  
  51.         }  
  52.     }  
  53.     return 0;  

좋은 웹페이지 즐겨찾기