bzoj 3307 雨天的尾巴

题目链接：传送门

题目大意：中文题，略

题目思路：网上有题解说是合并线段树的，但是太难蒟蒻不会，只能用树剖求解

　　　　　如果不是树而是一维数组我们会怎么解？

　　　　　当然是利用前缀和思想标记 （L） v+1，（R+1） v-1，然后扫一遍

　　　　　用线段树取最大复杂度 nlogn

　　　　　现在是搬到了树上，怎么做？

　　　　　利用树链剖分拆成链然后在像一维那样做就行，复杂度n(logn)^2

 #include <iostream>
 #include <cstdio>
 #include <cstdlib>
 #include <cmath>
 #include <algorithm>
 #include <cstring>
 #include <stack>
 #include <cctype>
 #include <queue>
 #include <string>
 #include <vector>
 #include <set>
 #include <map>
 #include <climits>
 #define lson rt<<1,l,mid
 #define rson rt<<1|1,mid+1,r
 #define fi first
 #define se second
 #define ping(x,y) ((x-y)*(x-y))
 #define mst(x,y) memset(x,y,sizeof(x))
 #define mcp(x,y) memcpy(x,y,sizeof(y))
 using namespace std;
 #define gamma 0.5772156649015328606065120
 #define MOD 1000000007
 #define inf 0x3f3f3f3f
 #define N 100005
 #define maxn 30010
 typedef pair<int,int> PII;
 typedef long long LL;
 LL read(){
     LL x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }
 vector<int>V[N];
 map<int,int>M;
 int num[N],ncnt,head[N],hcnt,ans[N];
 int n,m,k,L,R,sz,v;
 int son[N],siz[N],id[N],tid;
 int top[N],dep[N],fa[N],posi[N];
 struct Node{int to,nxt,v;}node[N<<];
 struct Seg{int v,va;}seg[N<<];
 inline void add(int x,int y,int v){
     node[hcnt].to=y,node[hcnt].nxt=head[x],node[hcnt].v=v,head[x]=hcnt++;
 }
 void dfs1(int u,int f,int deep){
     fa[u]=f,++siz[u],dep[u]=deep;
     for(int e:V[u]){
         if(e==f)continue;
         dfs1(e,u,deep+);siz[u]+=siz[e];
         if(!son[u]||siz[son[u]]<siz[e])son[u]=e;
     }
 }
 void dfs2(int u,int tp){
     id[u]=++tid,posi[tid]=u,top[u]=tp;
     if(!son[u])return; dfs2(son[u],tp);
     for(int e:V[u])if(!id[e])dfs2(e,e);
 }
 void pushup(int rt){
     if(seg[rt<<].v>seg[rt<<|].v)seg[rt]=seg[rt<<];
     else if(seg[rt<<].v<seg[rt<<|].v)seg[rt]=seg[rt<<|];
     else if(num[seg[rt<<].va]<num[seg[rt<<|].va])seg[rt]=seg[rt<<];
     else seg[rt]=seg[rt<<|];
 }
 void update(int rt,int l,int r){
     if(l==r){seg[rt].v+=v;seg[rt].va=l;return;}
     int mid=l+r>>;
     if(L<=mid)update(lson);
     else update(rson);
     pushup(rt);
 }
 void build(int rt,int l,int r){
     if(l==r){seg[rt].va=l;return;}
     int mid=l+r>>;
     build(lson);build(rson);
 }
 int main(){
     int i,j,group,x,y,Case=;
     n=read(),m=read();
     mst(head,-);
     for(i=;i<n;++i){
         x=read(),y=read();
         V[x].push_back(y);
         V[y].push_back(x);
     }
     dfs1(,,);dfs2(,);
     for(i=;i<=m;++i){
         x=read(),y=read();v=read();
         if(!M[v]){
             M[v]=++ncnt;
             num[ncnt]=v;
         }
         v=M[v];
         while(top[x]!=top[y]){
             if(dep[top[x]]<dep[top[y]])swap(x,y);
             L=id[top[x]],R=id[x];
             add(L,v,);add(R+,v,-);
             x=fa[top[x]];
         }
         if(dep[x]<dep[y])swap(x,y);
         L=id[y],R=id[x];
         add(L,v,);add(R+,v,-);
     }
     build(,,m);
     for(i=;i<=n;++i){
         for(j=head[i];~j;j=node[j].nxt){
             L=node[j].to;v=node[j].v;
             update(,,m);
         }
         ans[posi[i]]=num[seg[].va];
     }
     for(i=;i<=n;++i)printf("%d\n",ans[i]);
     return ;
 }