传送门
题解:先进行点分治,得到不同情况下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;
}