HDU-1698 Just a Hook 线段树染色

       题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1698

       线段树典型的染色操作

   My code:

//STATUS:C++_AC_359MS_3328KB  
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
#define LL __int64
const int MAX=100010,INF=200000000;
int min(int x,int y){return x<y?x:y;}
struct Tree{
    int l,r,num;
}tree[MAX*4];
void build(int l,int r,int k);
void update(int l,int r,int k,int z);
void search(int l,int r,int k);
int T,n,m,ans;
int main()
{
//    freopen("in.txt","r",stdin);
    int i,x,y,z,k=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        build(1,n,1);
        for(i=0;i<m;i++){
            scanf("%d%d%d",&x,&y,&z);
            update(x,y,1,z);
        }

        ans=0;
        search(1,n,1);
        printf("Case %d: The total value of the hook is %d.\n",k++,ans);
    }
    return 0;
}

void build(int l,int r,int k)
{
    tree[k].l=l,tree[k].r=r;
    tree[k].num=1;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(l,mid,k<<1);
    build(mid+1,r,(k<<1)+1);
}

void update(int l,int r,int k,int z)
{
    if(tree[k].l==l && tree[k].r==r || tree[k].num==z){
        tree[k].num=z;
        return;
    }
    if(tree[k].num!=z && tree[k].num){
        tree[k<<1].num=tree[k].num;
        tree[(k<<1)+1].num=tree[k].num;
        tree[k].num=0;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid)
        update(l,r,k<<1,z);
    else if(l>mid)
        update(l,r,(k<<1)+1,z);
    else {
        update(l,mid,k<<1,z);
        update(mid+1,r,(k<<1)+1,z);
    }
}

void search(int l,int r,int k)
{
    if(tree[k].l==l && tree[k].r==r && tree[k].num){
        ans+=tree[k].num*(r-l+1);
        return;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid)
        search(l,r,k<<1);
    else if(l>mid)
        search(l,r,(k<<1)+1);
    else {
        search(l,mid,k<<1);
        search(mid+1,r,(k<<1)+1);
    }
}