POJ1741 Tree

来源：互联网发布：有关打击电信网络诈骗编辑：程序博客网时间：2024/06/05 02:18

题意：找出一棵树中有多少个点对之间的距离小于等于k。。

论文上的题。

点分治，找到树的重心，然后计算经过这个重心的点对，计算从根到这个子树每个节点的距离，然后找到所有点对过根的距离小于等于k的，然后再减去再同一棵子树里的就行了，然后分治一直递归下去。。

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int MAXN=10010;const int INF=1<<30;typedef long long ll;bool vis[MAXN];struct EDGE{int v,next;int val;}edge[MAXN<<1];int head[MAXN],size;ll S[MAXN];void init(){memset(head,-1,sizeof(head));size=0;}void add_edge(int u,int v,int c){edge[size].v=v;edge[size].val=c;edge[size].next=head[u];head[u]=size++;}int num[MAXN],siz[MAXN],root;ll dep[MAXN],k;int tot_size;void get_root(int u,int fa)//找树的重心{siz[u]=1;num[u]=0;for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].v;if(vis[v]||v==fa)continue;get_root(v,u);siz[u]+=siz[v];num[u]=max(num[u],siz[v]);}num[u]=max(num[u],tot_size-num[u]);if(num[root]>num[u])root=u;}int cnt;void get_dep(int u,int fa)//每个点离重心的距离{S[cnt++]=dep[u];for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].v;if(v==fa||vis[v])continue;dep[v]=dep[u]+edge[i].val;get_dep(v,u);}}ll get_num(int u,int fa,int len)//计算{cnt=0;dep[u]=len;get_dep(u,fa);sort(S,S+cnt);ll l=0,r=cnt-1;ll res=0;while(l<r){if(S[r]+S[l]<=k){res+=r-l;l++;}elser--;}return res;}ll ans;void dfs(int u,int fa){ans+=get_num(u,fa,0);//找到经过重心的路径vis[u]=1;for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].v;if(vis[v])continue;ans-=get_num(v,u,edge[i].val);//减去不经过路径的重心root=0;tot_size=siz[v];memset(num,0,sizeof(num));num[0]=INF;get_root(v,-1);dfs(root,-1);}}int main(){int n,i;while(scanf("%d%I64d",&n,&k)==2){if(n==0&&k==0)break;int u,v,c;init();for(i=1;i<n;i++){scanf("%d%d%d",&u,&v,&c);add_edge(u,v,c);add_edge(v,u,c);}memset(num,0,sizeof(num));memset(vis,0,sizeof(vis));num[0]=INF;tot_size=n;root=0;get_root(1,-1);ans=0;dfs(root,-1);printf("%I64d\n",ans);}return 0;}

0 0