题意:给定一棵N(1<= N <=10000)个结点的带权树,定义dist(u,v)为u,v两点间的最短路径长度,路径的长度定义为路径上所有边的权和。再给定一个 K ,如果对于不同的两个结点a,b,如果满足dist(a,b) <=K,则称(a,b)为合法点对。求合法点对个数。
思路:看了论文《分治算法在树的路径问题中的应用》,里面讲解的很清楚,一条路径要么过根节点,要么在一颗子树中,所以用分治算法。找到树的重心作为根节点,这样每次树的节点数至少减少一半。处理经过当前根节点路径<=k的点对数,然后把根节点去掉后就把原来的树分成几颗子树了,再处理子树。我们在求经过一个根节点的路径时,里面还包含了点对属于同一颗子树的情况,所以要去掉这部分的点。
dis(i)+dis(j)<=k(i,j的父节点不为根节点的同一个儿子)
=dis(i)+dis(j)<=k-dis(i)+dis(j)<=k(i,j的父节点属于根节点的同一儿子).
#include <algorithm> #include<stdio.h> #include<string.h> const int N=10010; using namespace std; int head[N],num,f[N],son[N],n,D,root,size,ans,dis[N],d[N],cum; bool vis[N]; #define max(a,b) (a<b?b:a) struct edge { int st,ed,w,next; }e[N*2]; void addedge(int x,int y,int w) { e[num].st=x;e[num].ed=y;e[num].w=w;e[num].next=head[x];head[x]=num++; e[num].st=y;e[num].ed=x;e[num].w=w;e[num].next=head[y];head[y]=num++; } void getroot(int u,int father)//求树的重心 { int i,v; f[u]=0;son[u]=1; for(i=head[u];i!=-1;i=e[i].next) { v=e[i].ed; if(vis[v]||v==father)continue; getroot(v,u); son[u]+=son[v]; f[u]=max(f[u],son[v]); } f[u]=max(f[u],size-son[u]); if(f[u]<f[root])root=u; } void getdis(int u,int father)//求节点到根节点的距离 { int i,v; son[u]=1;//更新子树的节点的子节点数,不更新也能ac d[cum++]=dis[u];//将点到根节点的距离加入数组 for(i=head[u];i!=-1;i=e[i].next) { v=e[i].ed; if(vis[v]||v==father)continue; dis[v]=dis[u]+e[i].w; getdis(v,u); son[u]+=son[v]; } } int cont(int u,int mit) { int res=0,L,R; dis[u]=mit; cum=0; getdis(u,0); sort(d,d+cum);//将点到根节点的距离排序 for(L=0,R=cum-1;L<R;) { if(d[L]+d[R]<=D)//如果d[L]+d[R]<=D,L代表的节点可以与(R-L)个节点成对 res+=(R-L++); else R--; } return res; } void work(int u) { int i,v; vis[u]=true; ans+=cont(u,0);//路径经过该根节点的点对数 for(i=head[u];i!=-1;i=e[i].next) { v=e[i].ed; if(vis[v])continue; ans-=cont(v,e[i].w);//减去属于v子树的点对数 root=0;f[root]=size=son[v]; getroot(v,0);//求v子树的根节点 work(root);//求v子树的点对 } } int main() { int i,x,y,w; while(scanf("%d%d",&n,&D),n||D) { memset(head,-1,sizeof(head)); num=0; for(i=1;i<n;i++) { scanf("%d%d%d",&x,&y,&w); addedge(x,y,w); } memset(vis,false,sizeof(vis)); root=0;f[root]=size=n;ans=0; getroot(1,0); work(root); printf("%d ",ans); } return 0; }