【HDU1698】Just a Hook

单一标记线段树

#include<cstdio>
#include<cstring>
#define lson i<<1,l,mid
#define rson i<<1|1,mid+1,r
using namespace std;
const int N=400020;
int sum[N],tag[N];
void work(int),modify(int,int,int,int,int,int),
    add(int,int,int,int),release(int,int,int),
    build(int,int,int),update(int);
int main(){
    int t;
    scanf("%d",&t);
    for (int i=1;i<=t;i++)
        work(i);
    return 0;
}
void work(int k){
    int n,q,x,y,z;
    scanf("%d %d",&n,&q);
    build(1,1,n);
    memset(tag,0,sizeof(tag));
    for (int i=1;i<=q;i++){
        scanf("%d %d %d",&x,&y,&z);
        modify(1,1,n,x,y,z);
    }
    printf("Case %d: The total value of the hook is %d.\n",k,sum[1]);
}
void build(int i,int l,int r){
    if (l==r){
        sum[i]=1;return;
    }
    int mid=(l+r)>>1;
    build(lson);build(rson);
    update(i);
}
void update(int i){
    sum[i]=sum[i<<1]+sum[i<<1|1];
}
void modify(int i,int l,int r,int L,int R,int d){
    if (L<=l&&r<=R){
        add(i,l,r,d);return;
    }
    release(i,l,r);
    int mid=(l+r)>>1;
    if (L<=mid) modify(lson,L,R,d);
    if (mid<R) modify(rson,L,R,d);
    update(i);
}
void add(int i,int l,int r,int d){
    tag[i]=d;
    sum[i]=(r-l+1)*d;
}
void release(int i,int l,int r){
    if (tag[i]){
        int mid=(l+r)>>1;
        add(lson,tag[i]);
        add(rson,tag[i]);
        tag[i]=0;
    }
}