BZOJ3784 : 树上的路径

树的点分治，在分治的时候将所有点到根的距离依次放入一个数组q中。

对于一棵子树里的点，合法的路径一定是q[L]..q[R]的某个数加上自己到重心的距离。

定义五元组(v,l,m,r,w)，表示当前路径长度为v，在[l,r]里选出最大值m，并加上w。

用大根堆维护这些五元组，每次取出v最大的元素，并扩展出[l,m-1]以及[m+1,r]两个状态，用线段树查询区间最大值。

时间复杂度$O(n\log^2n+m\log n)$。

 

#include<cstdio>
#include<queue>
using namespace std;
const int N=50010,M=800000;
int n,m,i,x,y,z,g[N],v[N<<1],w[N<<1],ok[N<<1],nxt[N<<1],ed;
int all,f[N],son[N],now;
int L,R,q[M],cnt,val[2100000],cp;
struct P{
  int v,l,m,r,w;
  P(){}
  P(int _v,int _l,int _m,int _r=0,int _w=0){v=_v,l=_l,m=_m,r=_r,w=_w;}
  inline bool operator<(const P&b)const{return v<b.v;}
}tmp,pre[M];
priority_queue<P>Q;
inline int merge(int x,int y){return q[x]>q[y]?x:y;}
void build(int x,int a,int b){
  if(a==b){val[x]=a;return;}
  int mid=(a+b)>>1;
  build(x<<1,a,mid),build(x<<1|1,mid+1,b);
  val[x]=merge(val[x<<1],val[x<<1|1]);
}
int ask(int x,int a,int b,int c,int d){
  if(c<=a&&b<=d)return val[x];
  int mid=(a+b)>>1;
  if(d<=mid)return ask(x<<1,a,mid,c,d);
  if(c>mid)return ask(x<<1|1,mid+1,b,c,d);
  return merge(ask(x<<1,a,mid,c,d),ask(x<<1|1,mid+1,b,c,d));
}
inline void extend(int l,int r,int w){
  if(l>r)return;
  int x=ask(1,1,cnt,l,r);
  Q.push(P(q[x]+w,l,x,r,w));
}
inline void add(int x,int y,int z){v[++ed]=y;w[ed]=z;ok[ed]=1;nxt[ed]=g[x];g[x]=ed;}
void findroot(int x,int y){
  son[x]=1,f[x]=0;
  for(int i=g[x];i;i=nxt[i])if(ok[i]&&v[i]!=y){
    findroot(v[i],x);
    son[x]+=son[v[i]];
    if(son[v[i]]>f[x])f[x]=son[v[i]];
  }
  if(all-son[x]>f[x])f[x]=all-son[x];
  if(f[x]<f[now])now=x;
}
void dfs(int x,int y,int dis){
  q[++cnt]=dis;
  pre[cp++]=P(L,R,dis);
  for(int i=g[x];i;i=nxt[i])if(ok[i]&&v[i]!=y)dfs(v[i],x,dis+w[i]);
}
void solve(int x){
  int i;L=cnt+1;
  for(i=g[x];i;i=nxt[i])if(ok[i])R=cnt,dfs(v[i],x,w[i]);
  pre[cp++]=P(L,cnt,0);
  for(i=g[x];i;i=nxt[i])if(ok[i]){
    ok[i^1]=0;
    f[0]=all=son[v[i]];
    findroot(v[i],now=0);
    solve(now);
  }
}
int main(){
  scanf("%d%d",&n,&m);
  for(ed=i=1;i<n;i++)scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
  f[0]=all=n;findroot(1,now=0);solve(now);
  build(1,1,cnt);
  for(i=0;i<cp;i++)extend(pre[i].v,pre[i].l,pre[i].m);
  while(m--){
    tmp=Q.top(),Q.pop(),printf("%d\n",tmp.v);
    extend(tmp.l,tmp.m-1,tmp.w);
    extend(tmp.m+1,tmp.r,tmp.w);
  }
  return 0;
}