【BZOJ4317】Atm的树

Description

Atm有一段时间在虐qtree的题目,于是,他满脑子都是tree,tree,tree……
于是,一天晚上他梦到自己被关在了一个有根树中,每条路径都有边权,一个神秘的声音告诉他,每个点到其他的点有一个距离(什么是距离不用说吧),他需要对于每个点回答:从这个点出发的第k小距离是多少;
如果atm不能回答出来,那么明天4019的闹钟将不会响,4019全寝可能就迟到了,所以atm希望你帮帮他。

Input

第一行,两个正整数n,k,表示树的点数,询问的是第几小距离;
第二~n行,每行三个正整数x,y,w,表示x和y之间有一条边,x为父亲,边权为w;

Output

n行, 每行一个数,第i行输出从i开始第k小距离

Sample Input

5 2
1 5 2
1 2 4
2 3 6
2 4 5

Sample Output

4
5
10
9
6

HINT

100% n<=15000, 边权在1~10之间,为了方便,保证1为根;K<=5000

题解:依旧是动态点分治。

统计第k大不太好搞,我们对于每个点都二分一下,变成求到一个点距离<=mid的点有多少个,然后就变成熟悉的题了。

我们对于每个点维护一棵线段树,记录它在点分树的子树中有多少个点到它的距离为x,同时为了去重,还要维护一个从它父亲中减去的版本。

时间复杂度$O(nlog_n^3)$。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=15010;
int n,m,N,tot,cnt,rt,mn;
int to[maxn<<1],next[maxn<<1],head[maxn],val[maxn<<1],siz[maxn],fa[maxn],md[20][maxn<<1],pos[maxn],vis[maxn];
int Log[maxn<<1],dep[maxn],r1[maxn],r2[maxn];
struct sag
{
int ls,rs,siz;
}s[maxn*100];
inline int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-') f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
inline void add(int a,int b,int c)
{
to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt++;
}
void getrt(int x,int fa)
{
int tmp=0,i;
siz[x]=1;
for(i=head[x];i!=-1;i=next[i]) if(!vis[to[i]]&&to[i]!=fa)
getrt(to[i],x),tmp=max(tmp,siz[to[i]]),siz[x]+=siz[to[i]];
tmp=max(tmp,tot-siz[x]);
if(tmp<mn) mn=tmp,rt=x;
}
void dfs(int x)
{
pos[x]=++pos[0],md[0][pos[0]]=dep[x];
for(int i=head[x];i!=-1;i=next[i]) if(!dep[to[i]]) dep[to[i]]=dep[x]+val[i],dfs(to[i]),md[0][++pos[0]]=dep[x];
}
void solve(int x)
{
vis[x]=1;
for(int i=head[x];i!=-1;i=next[i]) if(!vis[to[i]])
tot=siz[to[i]],mn=1<<30,getrt(to[i],x),fa[rt]=x,solve(rt);
}
inline int dis(int x,int y)
{
int a=pos[x],b=pos[y];
if(a>b) swap(a,b);
int k=Log[b-a+1];
return dep[x]+dep[y]-2*min(md[k][a],md[k][b-(1<<k)+1]);
}
void insert(int l,int r,int &x,int a)
{
if(!x) x=++tot;
s[x].siz++;
if(l==r) return ;
int mid=(l+r)>>1;
if(a<=mid) insert(l,mid,s[x].ls,a);
else insert(mid+1,r,s[x].rs,a);
}
int query(int l,int r,int x,int a,int b)
{
if(a>b) return 0;
if(!x||(a<=l&&r<=b)) return s[x].siz;
int mid=(l+r)>>1;
if(b<=mid) return query(l,mid,s[x].ls,a,b);
if(a>mid) return query(mid+1,r,s[x].rs,a,b);
return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);
}
inline int calc(int x,int mid)
{
int ret=0,y,z;
for(y=x;y;y=z)
{
z=fa[y];
if(z) ret-=query(0,N,r2[y],0,mid-dis(x,z));
ret+=query(0,N,r1[y],0,mid-dis(x,y));
}
return ret;
}
int main()
{
n=rd(),m=rd()+1;
int i,j,u,v,a,b,c,l,r,mid;
memset(head,-1,sizeof(head));
for(i=1;i<n;i++) a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c),N+=c;
dep[1]=1,dfs(1),tot=n,mn=1<<30,getrt(1,0),solve(rt);
for(i=2;i<(n<<1);i++) Log[i]=Log[i>>1]+1;
for(j=1;(1<<j)<(n<<1);j++) for(i=1;i+(1<<j)-1<(n<<1);i++) md[j][i]=min(md[j-1][i],md[j-1][i+(1<<(j-1))]);
for(i=1;i<=n;i++)
{
for(u=i;u;u=v)
{
v=fa[u];
if(v) insert(0,N,r2[u],dis(i,v));
insert(0,N,r1[u],dis(i,u));
}
}
for(i=1;i<=n;i++)
{
l=0,r=N+1;
while(l<r)
{
mid=(l+r)>>1;
if(calc(i,mid)>=m) r=mid;
else l=mid+1;
}
printf("%d\n",r);
}
return 0;
}

最新文章

  1. 跳转时候提示Attempt to present on while a presentation is in progress
  2. RecyclerView局部刷新那点事
  3. java对象占多少内存
  4. C#_数据库连接串的配置
  5. PHP 获取服务器详细信息【转】
  6. Rest文件下载
  7. == 和equals比较
  8. 从烙铁手到IT男
  9. 控制器view的延迟加载
  10. Sphinx 排序模式 SetSortMode
  11. Tornado模块分类和各模块之间的关系
  12. 类加载器与methodinterceptor接口
  13. hdu1023
  14. springmvc 之 深入核心研究
  15. 例子:js超级玛丽小游戏
  16. Android Studio 封装的类的继承
  17. TensorFlow-Slim image classification library:TensorFlow-Slim 图像分类库
  18. Ubuntu每次启动都显示System program problem detected
  19. DG Switch over
  20. JAVA EXAM3 复习提纲

热门文章

  1. 一些制作app的软件
  2. unity3d常用控件
  3. 使用httpModules做一些事
  4. WebApi接口传参不再困惑(4):传参详解
  5. Eclipse - JAR包制作
  6. 阿里云slb实现多域名https
  7. C#通过WIN32 API实现嵌入程序窗体
  8. 在Unity控制台下使用富文本
  9. mybatis-config.xml文件详解
  10. atitit.验证码识别step3----去除边框---- 图像处理类库 attilax总结java版本