bzoj 3784（点分治+RMQ+堆）

传送门
题解：先进行点分治，得到不同情况下p点的dis[p]，以及当前情况下p点可以到达的子树区间，然后建st表，按NOI2010超级钢琴（昨天才做过。。。）那种方式维护一个优先队列即可。

注意：st表数组两维的先后很有讲究，n的那一维放后面，logn的那一维放前面，这样可以大幅减小高位寻址带来的时间复杂度。本人亲测，快了一倍！！！（7900ms–>3400ms）

#include<bits/stdc++.h>
using namespace std;
#define mp(a,b) make_pair(a,b)
typedef long long ll;
const int MAXN=5e4+5,MAXP=MAXN*40;
int n,m,siz[MAXN],mx[MAXN],tot=0,sum,root,l,r;
bool vis[MAXN];
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt,w;
}e[MAXN<<1];
int dis[MAXP],f[23][MAXP],lg[MAXP];
pair<int,int> po[MAXP];
struct NODE {
    int i,l,r,t;
    friend bool operator <(const NODE &a,const NODE &b) {
        return dis[a.i]+dis[a.t]<dis[b.i]+dis[b.t];
    }
};
priority_queue<NODE> q;
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
inline void adde(int u,int v,int w) {
    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;
}
void getroot(int p,int fa) {
    siz[p]=1,mx[p]=0;
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa&&!vis[v]) {
            getroot(v,p);
            siz[p]+=siz[v];
            mx[p]=max(mx[p],siz[v]);
        }
    }
    mx[p]=max(mx[p],sum-siz[p]);
    if (mx[p]<mx[root]) root=p;
}
void getdis(int p,int fa,int d) {
    siz[p]=1,dis[++tot]=d,po[tot]=mp(l,r);
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa&&!vis[v]) {
            getdis(v,p,e[i].w+d);
            siz[p]+=siz[v];
        }
    }
}
void work(int pos) {
    dis[l=r=++tot]=0,po[tot]=mp(0,0);
    vis[pos]=true;
    for (int i=head[pos];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (!vis[v]) {
            getdis(v,pos,e[i].w);
            r=tot;
        }
    }
    for (int i=head[pos];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (!vis[v]) {
            sum=siz[v],root=0;
            getroot(v,0);
            work(root);
        }
    }
}
void init_st() {
    lg[1]=0,f[0][1]=1;
    for (register int i=2;i<=tot;++i) lg[i]=lg[i>>1]+1,f[0][i]=i;
    for (int j=1;(1<<j)<=tot;++j)
        for (int i=1;i+(1<<j)-1<=tot;++i) {
            int x=f[j-1][i],y=f[j-1][i+(1<<(j-1))];
            f[j][i]=dis[x]>dis[y]?x:y;
        }
}
int rmq(int l,int r) {
    int t=lg[r-l+1];
    int x=f[t][l],y=f[t][r-(1<<t)+1];
    return dis[x]>dis[y]?x:y;
}
int main() {
//  freopen("bzoj 3784.in","r",stdin);
    memset(head,-1,sizeof(head));
    memset(vis,false,sizeof(vis));
    n=read(),m=read();
    for (register int i=1;i<n;++i) {
        int u=read(),v=read(),w=read();
        adde(u,v,w); 
    }
    sum=n,root=0,mx[0]=MAXN;
    getroot(1,0);
    work(root);
    init_st();
    for (register int i=1;i<=tot;++i) {
        if (!po[i].first) continue;
        int pos=rmq(po[i].first,po[i].second);
        q.push((NODE){i,po[i].first,po[i].second,pos});
    }
    for (register int b=0;b<m;++b) {
        NODE cur=q.top();q.pop();
        printf("%d\n",dis[cur.i]+dis[cur.t]);
        if (cur.l<cur.t) q.push((NODE){cur.i,cur.l,cur.t-1,rmq(cur.l,cur.t-1)});
        if (cur.t<cur.r) q.push((NODE){cur.i,cur.t+1,cur.r,rmq(cur.t+1,cur.r)});
    }
    return 0;
}