HDu 1698セグメント変更値

1482 ワード

hdu 1698 Just a hook
#include<stdio.h>
#include<string.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 100010
int sum[maxn<<2];
int col[maxn<<2];
int n,q;
void pushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    col[rt]=0;
    sum[rt]=1;
	if(l==r)
		return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}

void pushDown(int l,int r,int rt)
{
    if(col[rt])
    {
        col[rt<<1]=col[rt<<1|1]=col[rt];
        int m=r-l+1;
		sum[rt<<1]=(m-(m>>1))*col[rt];
        sum[rt<<1|1]=(m>>1)*col[rt];
        col[rt]=0;
    }
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        col[rt]=c;
        sum[rt]=c*(r-l+1);
        return;
    }
    pushDown(l,r,rt);
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,c,lson);
    if(m<R)
        update(L,R,c,rson);
    pushUp(rt);
}
int main()
{
    int i,j,k,l,ll;
    scanf("%d",&ll);
    for(l=1;l<=ll;l++)
    {
        scanf("%d %d",&n,&q);
        build(1,n,1);
        for(i=0;i<q;i++)
        {
            int a,b,c;
            scanf("%d %d %d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
        printf("Case %d: The total value of the hook is %d.
",l,sum[1]); } return 0; }