hdu_5029_relief grain(树链剖分)

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

题意：给你一个树，然后给你两点，将这两点之间的点涂上颜色，问涂色最多的那个颜色是什么，如果数量相同，就输出编号小的颜色

题解：树链剖分，这个题很巧妙，线段树维护的是一个颜色线段树，然后如果要涂[l,r]的颜色a，就在l处标记a +1，在r+1处标记a  -1。然后从左往右依次更新线段树，并query，就能得出答案，我的跑了1400+ms,鸟神的非递归线段树只跑我一半的时间

详解请看：鸟神的这题题解：relief grain

 #include<cstdio>
 #include<cstring>
 #define F(i,a,b) for(int i=a;i<=b;++i)
 #define root 0,100001,1
 #define ls l,m,rt<<1
 #define rs m+1,r,rt<<1|1
 #pragma comment(linker, "/STACK:102400000,102400000")
 const int N=;
 int n,m,x,y,z,idx,ans[N],sum[N*],nxt[*N],g[N],v[N*],w[N*],ed,pre[N],siz[N],dep[N],hson[N],top[N],tid[N],ftid[N];

 inline void adg(int x,int y,int z){v[++ed]=y,w[ed]=z,nxt[ed]=g[x],g[x]=ed;}
 //树链部分
 void dfs1(int u,int fa){
     pre[u]=fa,siz[u]=,dep[u]=dep[fa]+,hson[u]=;
     for(int i=g[u];~i;i=nxt[i]){
         int vv=v[i];
         if(vv!=fa){
             dfs1(vv,u);
             if(siz[vv]>siz[hson[u]])hson[u]=vv;
             siz[u]+=siz[vv];
         }
     }
 }

 void dfs2(int u,int tp){
     tid[u]=idx,ftid[idx++]=u,top[u]=tp;
     if(hson[u])dfs2(hson[u],tp);
     for(int i=g[u];~i;i=nxt[i]){
         int vv=v[i];
         if(vv!=pre[u]&&vv!=hson[u])dfs2(vv,vv);
     }
 }

 void Update(int u,int v,int c){
     int fu=top[u],fv=top[v];
     while(fu!=fv){
         if(dep[fu]>=dep[fv])
             adg(tid[fu],c,),adg(tid[u]+,c,-),u=pre[fu],fu=top[u];
         else adg(tid[fv],c,),adg(tid[v]+,c,-),v=pre[fv],fv=top[v];
     }
     if(dep[u]>dep[v])u=u^v,v=u^v,u=u^v;
     adg(tid[u],c,),adg(tid[v]+,c,-);
 }
 //线段树部分
 void update(int x,int v,int l,int r,int rt){
     if(l==r){sum[rt]+=v;return;}
     int m=(l+r)>>;
     if(x<=m)update(x,v,ls);else update(x,v,rs);
     sum[rt]=sum[rt<<]>sum[rt<<|]?sum[rt<<]:sum[rt<<|];
 }

 int query(int l,int r,int rt){
     if(l==r)return l;
     int m=(l+r)>>;
     if(sum[rt]==sum[rt<<])return query(ls);
     else return query(rs);
 }

 int main(){
     while(~scanf("%d%d",&n,&m),n+m){
         memset(g,-,sizeof(g)),ed=;
         F(i,,n-)scanf("%d%d",&x,&y),adg(x,y,),adg(y,x,);
         dfs1(,),idx=,dfs2(,);
         F(i,,*N-)sum[i]=;
         memset(g,-,sizeof(g)),ed=;
         F(i,,m)scanf("%d%d%d",&x,&y,&z),Update(x,y,z);
         F(u,,n){
             for(int i=g[u];~i;i=nxt[i])update(v[i],w[i],root);
             ans[ftid[u]]=query(root);
         }
         F(i,,n)printf("%d\n",ans[i]);
     }
     return ;
 }