zoukankan      html  css  js  c++  java
  • 线段树模板整理

      1 #include<cstdlib>
      2 #include<cstdio>
      3 #include<cmath>
      4 #include<cstring>
      5 #include<string>
      6 #include<iostream>
      7 #include<algorithm>
      8 #include<map>
      9 #include<queue>
     10 #include<vector>
     11 using namespace std;
     12 typedef long long LL;
     13 const LL MaxN = 1e5;
     14 typedef struct{
     15     int l, r;
     16     LL sum, lazy;
     17     void update(LL C)
     18     {
     19         sum += 1LL*(r-l+1)*C;
     20         lazy += C;
     21     }
     22 } Power;
     23 Power segtree[MaxN<<2];
     24 int n, q;
     25 int num[MaxN+5];
     26 
     27 void push_up(int x)
     28 {
     29     segtree[x].sum = segtree[x<<1].sum + segtree[x<<1|1].sum;
     30 }
     31 void push_down(int x)
     32 {
     33     int lazyval = segtree[x].lazy;
     34     if(lazyval){
     35         segtree[x<<1].update(lazyval);
     36         segtree[x<<1|1].update(lazyval);
     37         segtree[x].lazy = 0;
     38     }
     39 }
     40 
     41 //建树
     42 void Build(int x, int l, int r)
     43 {
     44     segtree[x].l = l, segtree[x].r = r;
     45     segtree[x].sum = segtree[x].lazy = 0;
     46     if(l == r){
     47         segtree[x].sum = num[l];
     48         return;
     49     }
     50     int mid = (l + r)/2;
     51     Build(x<<1, l, mid);
     52     Build(x<<1|1, mid+1, r);
     53     push_up(x);
     54 }
     55 
     56 //点修改,假设num[i]+C
     57 void Update_point(int x, int i, int C)
     58 {
     59     int l = segtree[x].l, r = segtree[x].r;
     60     if(l == r){
     61         segtree[x].sum += C;
     62         return;
     63     }
     64     int mid = (l+r)/2;
     65     if(i <= mid) Update_point(x<<1, i, C);
     66     else Update_point(x<<1|1, i, C);
     67     push_up(x);
     68 }
     69 
     70 //区间修改,假设num[L, R] += C
     71 void Update_segment(int x, int L, int R, LL C)
     72 {
     73     int l = segtree[x].l, r = segtree[x].r;
     74     if(l >= L && r <= R){
     75         segtree[x].update(C);
     76     }
     77     else{
     78         push_down(x);
     79         int mid = (l + r)/2;
     80         if(mid >= L) Update_segment(x<<1, L, R, C);
     81         if(mid < R) Update_segment(x<<1|1, L, R, C);
     82         push_up(x);
     83     }
     84 }
     85 
     86 //区间查询,假设查询num[L, R];
     87 LL Query_segment(int x, int L, int R)
     88 {
     89     int l = segtree[x].l, r = segtree[x].r;
     90     if(l >= L && r <= R){
     91         return segtree[x].sum;
     92     }
     93     else{
     94         push_down(x);
     95         LL ans = 0;
     96         int mid = (l + r)/2;
     97         if(mid >= L) ans += Query_segment(x<<1, L, R);
     98         if(mid < R) ans += Query_segment(x<<1|1, L, R);
     99         push_up(x);
    100         return ans;
    101     }
    102 }
    103 
    104 int main()
    105 {
    106     cin >> n;
    107     for(int i = 1;i <= n;i++){
    108         scanf("%d", num + i);
    109     }
    110     Build(1, 1, n);
    111     scanf("%d", &q);
    112     for(int i = 1;i <= q;i++){
    113         int l, r, C;
    114         scanf("%d %d %d", &l, &r, &C);
    115         Update_segment(1, l, r, C);
    116         printf("%lld
    ", Query_segment(1, l, r));
    117     }
    118     return 0;
    119 }
  • 相关阅读:
    Python学习-day10
    python学习-day9
    Python学习-day8
    Python学习-day7
    Python学习-day6
    Django2
    Django讲解
    JQuery讲解
    前端Day2
    前端Day1
  • 原文地址:https://www.cnblogs.com/Lightfall/p/9400582.html
Copyright © 2011-2022 走看看