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

来源:互联网 发布:大学英语听力软件推荐 编辑:程序博客网 时间:2024/04/30 09:26

题意


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 100010using 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;}
0 0
原创粉丝点击