hdu 1556 color the ball 线段树区间更新 加lazy标记

题意：给你一排气球让你涂色， 有n个操作，表示将[l，r]区间的球涂色，问最后1~n的气球被涂色的次数。

#include<iostream>
#include<cstdio>
#include<cstring>
const int MAXN = 100005;
using namespace std;
struct t
{
    int l, r, c;
} tree[MAXN*4];
int lazy[MAXN*4];//数组约4倍大，lazy标记，延迟更新
void pushup(int k)
{  tree[k].c=tree[k<<1].c+tree[k<<1|1].c;
}
void build_tree(int l,int r,int k)
{
    tree[k].l=l;
    tree[k].r=r;
    tree[k].c=0;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    build_tree(l, mid, k<<1);
    build_tree(mid+1, r, k<<1|1);
    //pushup(k);
}

void pushdown(int k)
{
    if(lazy[k])
    {
        lazy[k<<1]+=lazy[k];
        lazy[k<<1|1]+=lazy[k];
        tree[k<<1].c+=lazy[k]*(tree[k<<1].r-tree[k<<1].l+1);
        tree[k<<1|1].c+=lazy[k]*(tree[k<<1|1].r-tree[k<<1|1].l+1);
        lazy[k]=0;
    }
    return ;
}
void update(int k, int l, int r)
{
    if(tree[k].l>=l&&tree[k].r<=r)
    {
        lazy[k]++;
        tree[k].c+=(tree[k].r-tree[k].l+1);
        return ;
    }
    pushdown(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid)//要更新的区间在当前节点的左子树
        update(k<<1,l,r);
    else if(l>mid)//要更新的区间在当前节点的右子树
        update(k<<1|1,l,r);
    else//左右子树都有
    {
        update(k<<1,l,mid);
        update(k<<1|1,mid+1,r);
    }
    pushup(k);
}
int query(int k, int x)
{
    if(tree[k].l==tree[k].r)
    {
        return tree[k].c;
    }
    pushdown(k);
    int ans=0;
    int mid = (tree[k].l+tree[k].r)>>1;
    if(x<=mid)
        ans += query(k<<1, x);
    else
        ans += query(k<<1|1, x);
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n),n)
    {
        int l, r;
        build_tree(1,n,1);
        memset(lazy, 0, sizeof(lazy));
        for(int i=0; i<n; i++)
        {
            scanf("%d %d",&l,&r);
            update(1,l,r);
        }
        for(int i=1; i<=n; ++i)
        {
            printf("%d%c", query(1, i), i==n?'\n':' ');
        }
    }
    return 0;
}

写法二：

#include<iostream>
#include<cstdio>
#include<cstring>
const int MAXN = 100005;
using namespace std;
struct T
{
    int l, r, c;
} tree[MAXN*4];
int lazy[MAXN*4];
void build_tree(int l, int r, int k)
{
    tree[k].l=l;
    tree[k].r=r;
    tree[k].c=0;
    if(l==r)
        return ;
    int mid = (tree[k].l+tree[k].r)>>1;
    build_tree(l, mid, k<<1);
    build_tree(mid+1, r,k<<1|1);
}
void pushup(int k)
{
    tree[k].c=tree[k<<1].c+tree[k<<1|1].c;
}
void pushdown(int k)
{
    if(lazy[k]!=0)
    {
        tree[k*2].c+=lazy[k];
        tree[k*2+1].c+=lazy[k];
        lazy[k*2]+=lazy[k];
        lazy[k*2+1]+=lazy[k];
        lazy[k]=0;
    }
}
void update(int k,int l,int r)
{
    if(l<=tree[k].l&&r>=tree[k].r)
    {
        tree[k].c++;
        lazy[k]++;
        return ;
    }
    pushdown(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(l<=mid)
        update(k*2,l,r);
    if(r>mid)
        update(k*2+1,l,r);
}
int query(int k,int x)
{
    if(tree[k].l==tree[k].r)
    {
        return tree[k].c;
    }
    pushdown(k);
    int mid=(tree[k].l+tree[k].r)/2;
    if(x<=mid)
        return query(k*2,x);
    else
        return query(k*2+1,x);
}
int main()
{
    int n;
    while(scanf("%d",&n),n)
    {
        int l, r;
        memset(tree, 0, sizeof(tree));
        memset(lazy, 0, sizeof(lazy));
        build_tree(1, n, 1);
        for(int i=0; i<n; i++)
        {
            scanf("%d %d",&l,&r);
            update(1,l,r);
        }
        for(int i=1; i<=n; ++i)
        {
            printf("%d%c", query(1, i), i==n?'\n':' ');
        }
    }
}