感觉现在写点分治可快了~

二分答案,就可以将求第 $k$ 大转换成一个判断问题,直接拿点分树判断一下就行了.

#include <cstdio>
#include <vector>
#include <algorithm>
#define N 100004
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int n,edges,K;
int hd[N],to[N<<1],nex[N<<1],val[N<<1];
void add(int u,int v,int c)
{
nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c;
}
namespace tree
{
int size[N],son[N],fa[N],top[N],dep[N],dis[N];
void dfs1(int u,int ff)
{
fa[u]=ff,size[u]=1;
for(int i=hd[u];i;i=nex[i])
if(to[i]!=ff)
{
dep[to[i]]=dep[u]+1,dis[to[i]]=dis[u]+val[i];
dfs1(to[i],u);
size[u]+=size[to[i]];
if(size[to[i]]>size[son[u]]) son[u]=to[i];
}
}
void dfs2(int u,int tp)
{
top[u]=tp;
if(son[u]) dfs2(son[u],tp);
for(int i=hd[u];i;i=nex[i])
if(to[i]!=fa[u]&&to[i]!=son[u])
dfs2(to[i],to[i]);
}
int LCA(int x,int y)
{
while(top[x]!=top[y])
dep[top[x]]>dep[top[y]]?x=fa[top[x]]:y=fa[top[y]];
return dep[x]<dep[y]?x:y;
}
int Dis(int x,int y)
{
return dis[x]+dis[y]-(dis[LCA(x,y)]<<1);
}
};
vector<int>F[N],G[N];
int root,sn;
int mx[N],size[N],vis[N],Fa[N];
void dfs(int u,int ff)
{
size[u]=1;
for(int i=hd[u];i;i=nex[i])
if(!vis[to[i]]&&to[i]!=ff)
dfs(to[i],u),size[u]+=size[to[i]];
}
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 calc(int u,int ff,int dep,int rt)
{
F[rt].push_back(dep);
if(Fa[rt]) G[rt].push_back(tree::Dis(Fa[rt], u));
for(int i=hd[u];i;i=nex[i])
if(to[i]!=ff&&!vis[to[i]])
calc(to[i],u,dep+val[i],rt);
}
void prepare(int u)
{
vis[u]=1;
calc(u,0,0,u);
F[u].push_back(1000000004);
sort(F[u].begin(),F[u].end());
if(Fa[u])
{
G[u].push_back(1000000004);
sort(G[u].begin(),G[u].end());
}
for(int i=hd[u];i;i=nex[i])
if(!vis[to[i]])
dfs(to[i],u),sn=size[to[i]],root=0,getroot(to[i],u),Fa[root]=u,prepare(root);
}
int query(int u,int k)
{
int U=u, re=upper_bound(F[u].begin(),F[u].end(),k)-F[u].begin()-1;
for(;Fa[u];u=Fa[u])
{
int dis=tree::Dis(Fa[u],U);
if(dis<=k)
{
re+=(upper_bound(F[Fa[u]].begin(),F[Fa[u]].end(),k-dis)-F[Fa[u]].begin());
re-=(upper_bound(G[u].begin(),G[u].end(),k-dis)-G[u].begin());
}
}
return re;
}
int main()
{
int i,j,tot=0;
char ss[2];
// setIO("input");
scanf("%s%d%d",ss,&n,&K);
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;
}
tree::dfs1(1,0);
tree::dfs2(1,1);
mx[0]=sn=n,root=0,getroot(1,0),prepare(root);
for(i=1;i<=n;++i)
{
int l=1,r=tot,mid,ans=0;
while(l<=r)
{
mid=(l+r)>>1;
if(query(i,mid)>=K)
{
ans=mid,r=mid-1;
}
else l=mid+1;
}
printf("%d\n",ans);
}
return 0;
}

  

最新文章

  1. CI关于自动加载
  2. 微信公众平台开发(三) 订阅事件(subscribe)处理
  3. android-数据存储之手机内部file存储
  4. access的逻辑类型
  5. HDU 2149 (巴什博奕) Public Sale
  6. python django第二弹
  7. Swift - 使用网格(UICollectionView)的自定义布局实现复杂页面
  8. redis Web服务器
  9. Luogu P2148 [SDOI2009]E&amp;D
  10. 洛谷P1605:迷宫(DFS)
  11. STL 标准模板库
  12. learning ddr reset initialization with stable power
  13. 动手动脑-java重载
  14. css卷叶效果
  15. OpenGL中的原子操作需要注意的地方
  16. 【Linux】SVN的安装和配置
  17. 木马分析出现python语言,360的安全人员不禁感叹还有这种操作?
  18. svchost.exe占网速的解决办法
  19. 账号被锁无法ssh登陆
  20. 【ghost初级教程】 怎么搭建一个免费的ghost博客

热门文章

  1. Clion 常用功能
  2. 双01字典树最小XOR(three arrays)--2019 Multi-University Training Contest 5(hdu杭电多校第5场)
  3. jQuery 实现图片放大两种方式
  4. 计算机和python
  5. 基础数据类型之集合和深浅copy,还有一些数据类型补充
  6. redis 学习(5)-- 列表类型
  7. 多Y轴,下拉框渲染,相同类型不同数据
  8. 多个分组树分别实现单选jq+angular
  9. MyCat配置简述以及mycat全局ID
  10. Docker的bridge和macvlan两种网络模式