码迷,mamicode.com
首页 > 其他好文 > 详细

线段树模板

时间:2019-03-01 15:52:25      阅读:186      评论:0      收藏:0      [点我收藏+]

标签:push   sum   lazy   down   +=   turn   return   date   div   

void pushdown(int l, int r, int rt)
{
    if (lazy[rt])
    {
        sum[rt << 1] += l * lazy[rt];
        sum[(rt << 1) + 1] += r * lazy[rt];
        lazy[rt << 1] += lazy[rt];
        lazy[(rt << 1) + 1] += lazy[rt];
        lazy[rt] = 0;
    }
}
void build(int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] = tmp[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, rt << 1);
    build(mid + 1, r, (rt << 1) + 1);
    sum[rt] = sum[rt << 1] + sum[(rt << 1) + 1];
}
void update(int l, int r, int rt, int l1, int r1, int v)
{
    if (l1 <= l && r <= r1)
    {
        sum[rt] += v * (r - l + 1);
        lazy[rt] += v;
        return;
    }
    int mid = (l + r) >> 1;
    pushdown(mid - l + 1, r - mid, rt);
    if (l1 <= mid)
        update(l, mid, rt << 1, l1, r1, v);
    if (r1 > mid)
        update(mid + 1, r, (rt << 1) + 1, l1, r1, v);
    sum[rt] = sum[rt << 1] + sum[(rt << 1) + 1];
}
int query(int l, int r, int rt, int l1, int r1)
{
    if (l1 <= l && r <= r1)
        return (ll)sum[rt];
    int mid = (l + r) >> 1;
    int ret = 0;
    pushdown(mid - l + 1, r - mid, rt);
    if (l1 <= mid)
        ret += query(l, mid, rt << 1, l1, r1);
    if (r1 > mid)
        ret += query(mid + 1, r, (rt << 1) + 1, l1, r1);
    sum[rt] = sum[rt << 1] + sum[(rt << 1) + 1];
    return ret;
}

 

线段树模板

标签:push   sum   lazy   down   +=   turn   return   date   div   

原文地址:https://www.cnblogs.com/nublity/p/10456611.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!