HDU 1698 JUST A HOOK(线段树)

#include<stdio.h>
#include<string.h>
#define L(x)(x<<1)
#define R(x)(x<<1|1)

#define maxN 100010
int num;
struct node{
int l,r,k;
}po[4*maxN];

void Build_tree(int st,int l,int r)
{
po[st].l=l,po[st].r=r,po[st].k=1;
if(l==r)return ;
int mid=(l+r)>>1;
Build_tree(L(st),l,mid);
Build_tree(R(st),mid+1,r);
}
void add(int l,int r,int value,int st)
{
if(po[st].l==l&&po[st].r==r)
{
po[st].k=value;
return ;
}
if(po[st].k==value)return ;
if(po[st].k>0)po[R(st)].k=po[L(st)].k=po[st].k;
po[st].k=0;
int mid=(po[st].r+po[st].l)>>1;
if(r<=mid){
add(l,r,value,L(st));
}
else if(l>mid){
add(l,r,value,R(st));
}
else{
add(l,mid,value,L(st));
add(mid+1,r,value,R(st));
}
}
int get_sum(int st)
{
if(po[st].k!=0||po[st].r==po[st].l)
return po[st].k*(po[st].r-po[st].l+1);
return get_sum(L(st))+get_sum(R(st));
}

int main()
{
int t,n,m,x,y,z;
scanf("%d",&t);
for(int j=1;j<=t;j++)
{
printf("Case %d: The total value of the hook is ",j);
num=0;
scanf("%d%d",&n,&m);
Build_tree(1,1,n);
while(m--){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z,1);
}
printf("%d.\n",get_sum(1));
}
}