标签:
题意:求一棵树上任意两点之间距离小于k的点对有多少个
分析:树分治的练手题,找到重心以后,我们要求所有经过中心的点,如何求这样的点呢,显然,所有的路径可以分为过重心,不过中心,终点是重心,所有点到达重心的距离很好求,可以很容易的求出,这些边由两部分组成,来自不同的儿子和来自相同的儿子,只要减去来自相同的儿子的数量就是过重心的数量
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=2e4+5;
typedef long long ll;
typedef unsigned long long ull;
int vis[maxn],n,k,num,d[maxn],_min,tsize[maxn],tssize[maxn],root;
int head[maxn],next[maxn],edge[maxn],val[maxn],tot;
ll ans;
void init(){
memset(vis,0,sizeof(vis));
tot=ans=0;
memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int w){
edge[tot]=v;
val[tot]=w;
next[tot]=head[u];
head[u]=tot++;
}
//获取子树大小和子树的最大值
void getsize(int u,int f){
tsize[u]=1;tssize[u]=0;
for(int i=head[u];i!=-1;i=next[i]){
int v=edge[i];
if(v==f||vis[v])continue;
getsize(v,u);
tsize[u]+=tsize[v];
tssize[u]=max(tssize[u],tsize[v]);
}
}
//重心
void getroot(int r,int u,int f){
tssize[u]=max(tssize[u],tsize[r]-tsize[u]);
if(tssize[u]<_min)_min=tssize[u],root=u;
for(int i=head[u];i!=-1;i=next[i]){
int v=edge[i];
if(v==f||vis[v])continue;
getroot(r,v,u);
}
}
//求所有点到u的距离
void getdis(int u,int dcnt,int f){
d[num++]=dcnt;
for(int i=head[u];i!=-1;i=next[i]){
int v=edge[i];int w=val[i];
if(v==f||vis[v])continue;getdis(v,dcnt+w,u);
}
}
int cal(int u,int dcnt){
num=0;int ret=0;
getdis(u,dcnt,-1);
int i=0,j=num-1;
sort(d,d+num);
//for(int i=0;i<num;i++)cout<<d[i]<<" ";cout<<endl;
while(i<j){
while(d[i]+d[j]>k&&i<j)j--;
ret+=j-i;
i++;
}
//cout<<ret<<endl;
return ret;
}
void dfs(int u){
_min=n;
getsize(u,-1);
getroot(u,u,-1);//cout<<root<<endl;
ans+=cal(root,0);
//cout<<root<<" "<<ans<<endl;
vis[root]=1;
int qqq=root;
for(int i=head[qqq];i!=-1;i=next[i]){
int v=edge[i],w=val[i];
if(vis[v])continue;
ans-=cal(v,w);
dfs(v);
}
//cout<<root<<" "<<ans<<endl;
}
int main(){
//freopen("in","r",stdin);
ios::sync_with_stdio(false);
while(cin>>n>>k&&n+k){
init();
for(int i=0;i<n-1;i++){
int u,v,w;cin>>u>>v>>w;
addedge(u,v,w);addedge(v,u,w);
}
dfs(1);
cout<<ans<<endl;
}
return 0;
}
标签:
原文地址:http://www.cnblogs.com/jihe/p/5954284.html