很容易想出二分这个思路,但是要想办法去掉一个 $log$.
没错,空间换时间.
双指针的部分错了好几次~
Code:
#include <set>
#include <queue>
#include <cstdio>
#include <vector>
#include <algorithm>
#define N 2000003
#define ll long long
#define setIO(s) freopen(s".in","r",stdin) , freopen(s".out","w",stdout)
using namespace std;
ll answer;
priority_queue<int>q;
vector<int>G[N],F[N],go[N];
int edges,n,root,sn,tmp,len,tt,m,flag;
int hd[N],nex[N<<1],to[N<<1],val[N<<1];
int size[N],mx[N],vis[N];
bool cmp(int a,int b)
{
return a>b;
}
void add(int u,int v,int c)
{
nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c;
}
void getroot(int u,int ff)
{
size[u]=1,mx[u]=0;
for(int i=hd[u];i;i=nex[i])
if(to[i]!=ff&&!vis[to[i]])
getroot(to[i],u),size[u]+=size[to[i]],mx[u]=max(mx[u],size[to[i]]);
mx[u]=max(mx[u],sn-size[u]);
if(mx[u]<mx[root]) root=u;
}
void getdis(int u,int ff,int dep)
{
F[tmp].push_back(dep);
for(int i=hd[u];i;i=nex[i])
if(to[i]!=ff&&!vis[to[i]])
getdis(to[i],u,dep+val[i]);
}
void prepare(int u)
{
vis[u]=1,G[u].push_back(++tmp),getdis(u,0,0);
for(int i=hd[u];i;i=nex[i])
if(!vis[to[i]])
{
G[u].push_back(++tmp),getdis(to[i],u,val[i]);
root=0,sn=size[to[i]],getroot(to[i],u),go[u].push_back(root),prepare(root);
}
}
ll calc(int u,int t)
{
int l=0,r=F[u].size()-1,pre=0;
if(r<=0) return 0;
ll sum=((ll)(r+1)*r)/2;
while(l<r)
{
if(F[u][l]+F[u][r]<t)
{
sum-=(ll)(r-l);
--r;
}
else
{
++l;
}
}
// printf("%lld\n",sum);
return sum;
}
void dfs(int u,int t)
{
int i;
if(!G[u].empty()) answer+=calc(G[u][0],t);
for(i=1;i<(int)G[u].size();++i) answer-=calc(G[u][i],t);
for(i=0;i<(int)go[u].size();++i) dfs(go[u][i],t);
}
ll check(int t)
{
answer=0,dfs(tt,t);
return answer;
}
multiset<int>S;
multiset<int>::iterator it;
// 长度大于 len 即可
void solve(int u)
{
int i,j;
if(!G[u].empty())
{
for(j=0;j<F[G[u][0]].size();++j)
{
if(F[G[u][0]][j]>len) q.push(F[G[u][0]][j]);
}
if(G[u].size()>=2)
{
int t=G[u][1];
for(j=0;j<F[t].size();++j) S.insert(-F[t][j]);
}
}
for(i=2;i<G[u].size();++i)
{
int t=G[u][i];
// F[t] 与 S 结合
it=S.begin();
for(j=0;j<F[t].size()&&(-(*S.begin())+F[t][0])>len;++j)
{
for(it=S.begin();it!=S.end();it++)
{
if(-(*it)+F[t][j]>len)
{
q.push(-(*it)+F[t][j]);
}
else break;
}
}
for(j=0;j<F[t].size();++j)
S.insert(-F[t][j]);
}
S.clear();
for(int i=0;i<go[u].size();++i) solve(go[u][i]);
}
int main()
{
int i,j,tot=0;
// setIO("input");
scanf("%d%d",&n,&m);
for(i=1;i<n;++i)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c),add(a,b,c),add(b,a,c),tot+=c;
}
mx[root=0]=sn=n,getroot(1,0),tt=root,prepare(root);
for(i=1;i<=tmp;++i) sort(F[i].begin(),F[i].end(),cmp);
int l=0,r=tot,mid,ans=0;
while(l<=r)
{
mid=(l+r)>>1;
if(check(mid)>=1ll*m) ans=mid,l=mid+1;
else r=mid-1;
}
len=ans,solve(tt);
while(q.size()<m) q.push(len);
while(!q.empty()) printf("%d\n",q.top()), q.pop();
return 0;
}