[BZOJ3307][线段树合并]雨天的尾巴

题意

---

N个点，形成一个树状结构。
有M次发放，每次选择两个点x,y 对于x到y的路径上（含x,y)每个点发一袋Z类型的物品。
完成所有发放后，每个点存放最多的是哪种物品。

---

看POPOQQQ大神的题解吧……
http://blog.csdn.net/popoqqq/article/details/45674373

之前觉得线段树合并没什么用，现在打脸了……

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#define N 100010

using namespace std;

typedef pair<int,int> inint;

int n,m,cnt,k,w;
int Ans[N],G[N],h[N],dpt[N],rt[N],p[N];
int fa[N][21];
struct edge{
  int t,nx;
}E[N<<1];
struct seg{
  int l,r,ls,rs,Max,tms;
}T[N*40];
struct stp{
  int u,v,w;
  friend bool operator <(stp a,stp b){
    return a.w<b.w;
  }
}A[N];
vector<int> Dec[N],Inc[N];

inline void reaD(int &x){
  char c=getchar();x=0;
  for(;c<48||c>57;c=getchar());for(;c>=48&&c<=57;x=x*10+c-48,c=getchar());
}

inline void Insert(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
}

void explore(int x,int f){
  fa[x][0]=f; dpt[x]=dpt[f]+1;
  for(int i=1;i<=20;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=f) explore(E[i].t,x);
}

inline int LCA(int x,int y){
  if(dpt[x]<dpt[y]) return LCA(y,x);
  for(int i=20;~i;i--)
    if(dpt[fa[x][i]]>=dpt[y]) x=fa[x][i];
  if(x==y) return x;
  for(int i=20;~i;i--)
    if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}

inline void update(int g){
  int a=T[T[g].ls].tms,b=T[T[g].rs].tms;
  T[g].tms=max(a,b);
  if(a>b) T[g].Max=T[T[g].ls].Max;
  else if(a<b) T[g].Max=T[T[g].rs].Max;
  else T[g].Max=min(T[T[g].ls].Max,T[T[g].rs].Max);
}

void merge(int &x,int y){
  if(x==0||y==0) return (void)(x=x+y);
  if(T[x].l==T[x].r) return (void)(T[x].tms+=T[y].tms);
  merge(T[x].ls,T[y].ls);
  merge(T[x].rs,T[y].rs);
  update(x);
}

void update(int &g,int x,int l,int r,int w){
  if(!g){
    g=++cnt;
    T[g].l=l; T[g].r=r;
    T[g].ls=T[g].rs=0;
  }
  if(T[g].l==T[g].r) return (void)(T[g].Max=l,T[g].tms+=w);
  int mid=l+r>>1;
  if(x<=mid) update(T[g].ls,x,l,mid,w);
  else update(T[g].rs,x,mid+1,r,w);
  update(g);    
}

void dfs(int x){
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=fa[x][0]){
      dfs(E[i].t);
      merge(rt[x],rt[E[i].t]);
    }
  for(int i=0;i<Inc[x].size();i++)
    update(rt[x],Inc[x][i],1,k,1);
  for(int i=0;i<Dec[x].size();i++)
    update(rt[x],Dec[x][i],1,k,-1);
  Ans[x]=p[T[rt[x]].Max];
}

int main(){
  freopen("3307.in","r",stdin);
  freopen("3307.out","w",stdout);
  reaD(n); reaD(m);
  for(int i=1;i<n;i++){
    int u,v;
    reaD(u); reaD(v);
    Insert(u,v);
  }
  explore(1,0);
  for(int i=1;i<=m;i++)
    reaD(A[i].u),reaD(A[i].v),reaD(A[i].w);
  sort(A+1,A+1+m);
  k=0,w=-1;
  for(int i=1;i<=m;i++){
    if(A[i].w!=w) p[++k]=A[i].w,w=A[i].w;
    int par=LCA(A[i].u,A[i].v);
    Inc[A[i].u].push_back(k);
    Inc[A[i].v].push_back(k);
    Dec[par].push_back(k);
    Dec[fa[par][0]].push_back(k);
  }
  cnt=0; dfs(1);
  for(int i=1;i<=n;i++)
    printf("%d\n",Ans[i]);
  return 0;
}