线段树 模板

对于二叉树来说,2n是左子节点,2n+1是右子节点。

建树复杂度为O(n),更新和查询的复杂度都是O(lgn)。

要注意对一个区间进行更新的时候不需要一直更新到底,可以提高效率,需要的时候往下带就行。

#define maxn 200000

struct node{
    int l, r, sum;
    int mid(){
        return (l+r)>>1;
    }
};

node Tree[maxn<<2];
int value[maxn+10]; //从 1 开始赋值 

//初始化树,根节点是1
void init_tree(int root, int l, int r){
    Tree[root].l = l;
    Tree[root].r = r;
    if(l == r)
        Tree[root].sum = value[l];
    else{
        init_tree(root<<1, l, (l+r)>>1);
        init_tree((root<<1)+1, ((l+r)>>1) + 1, r);

        Tree[root].sum = Tree[root<<1].sum + Tree[(root<<1)+1].sum;
    }
}

//查找和
int query_tree(int root, int l, int r){
    int m = Tree[root].mid();
    if(l == Tree[root].l && r == Tree[root].r)
        return Tree[root].sum;
    else{
        if(l > m)
            return query_tree((root<<1)+1, l, r);
        else if(r <= m)
            return query_tree(root<<1, l, r);
        else
            return query_tree(root<<1, l, m) + query_tree((root<<1)+1, m+1, r);
    }
}

void update_tree(int root, int idx, int v){
    if(Tree[root].l == Tree[root].r)
        Tree[root].sum += v;
    else{
        if(idx <= Tree[root].mid())
            update_tree(root<<1, idx, v);
        else
            update_tree((root<<1)+1, idx, v);
           Tree[root].sum = Tree[root<<1].sum + Tree[(root<<1)+1].sum; 
    }
}

出处:http://blog.csdn.net/cfarmerreally/article/details/53574139


typedef long long ll;
const int maxn=50000+10; //数组大小

int lazy[maxn<<2]; //懒惰标记
int sum[maxn<<2];
int value[maxn]; //初始值

void PushUp(int rt) //向上更新父节点
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void PushDown(int rt,int m) //向下更新
{
    if(lazy[rt]) //懒惰标记,这里的语句根据更新函数而变
    {
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        sum[rt<<1]+=(m-(m>>1))*lazy[rt];
        sum[rt<<1|1]+=(m>>1)*lazy[rt];
        lazy[rt]=0;
    }
}

void build(int l,int r,int rt) //建树
{
    lazy[rt]=0;
    if(l==r)
    {
        sum[rt]=value[l];
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    PushUp(rt);
}

void update(int L,int R,int c,int l,int r,int rt) //将L~R区间加上c,可以根据需要修改更新函数
{
    if(L<=l&&R>=r)
    {
        lazy[rt]+=c;
        sum[rt]+=c*(r-l+1);
        return;
    }
    PushDown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,l,m,rt<<1);
    if(m<R) update(L,R,c,m+1,r,rt<<1|1);
    PushUp(rt);
}

ll query(int L,int R,int l,int r,int rt) //查询L~R
{
    if(L<=l&&R>=r)
    return sum[rt];
    PushDown(rt,r-l+1);
    int m=(l+r)>>1;
    ll ret=0;
    if(L<=m) ret+=query(L,R,l,m,rt<<1);
    if(m<R) ret+=query(L,R,m+1,r,rt<<1|1);
    return ret;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值