题意：
求树上的距离小于等于K的点对个数
题解：
树分治论文题。
需要者请移步qzc论文~
View Code
1 #include <iostream> 2 #include <cstring> 3 #include <algorithm> 4 #include <cstdio> 5 #include <cstdlib> 6 7 #define N 200000 8 #define M 400000 9 #define INF 0x3f3f3f3f 10 11 using namespace std; 12 13 int head[N],next[M],to[M],len[M]; 14 int sz[N],lim[N],dis[N]; 15 bool vis[N]; 16 int n,cnt,root,tot,mn,md; 17 18 inline void add(int u,int v,int w) 19 { 20 to[cnt]=v; len[cnt]=w; next[cnt]=head[u]; head[u]=cnt++; 21 } 22 23 inline void read() 24 { 25 memset(head,-1,sizeof head); cnt=0; 26 for(int i=1,a,b,c;i<n;i++) 27 { 28 scanf("%d%d%d",&a,&b,&c); 29 add(a,b,c); add(b,a,c); 30 } 31 } 32 33 inline void getsize(int u,int fa) 34 { 35 sz[u]=1; lim[u]=0; 36 for(int i=head[u];~i;i=next[i]) 37 if(to[i]!=fa&&!vis[to[i]]) 38 { 39 getsize(to[i],u); 40 sz[u]+=sz[to[i]]; 41 lim[u]=max(lim[u],sz[to[i]]); 42 } 43 } 44 45 inline void getroot(int p,int u,int fa) 46 { 47 lim[u]=max(lim[u],sz[p]-sz[u]); 48 if(lim[u]<mn) mn=lim[u],root=u; 49 for(int i=head[u];~i;i=next[i]) 50 if(to[i]!=fa&&!vis[to[i]]) getroot(p,to[i],u); 51 } 52 53 inline void getdis(int u,int fa,int d) 54 { 55 for(int i=head[u];~i;i=next[i]) 56 if(to[i]!=fa&&!vis[to[i]]) getdis(to[i],u,d+len[i]); 57 dis[++tot]=d; 58 } 59 60 inline int calc(int u,int fa,int d) 61 { 62 tot=0; 63 getdis(u,fa,d); 64 sort(dis+1,dis+1+tot); 65 int res=0; 66 for(int i=1,j=tot;i<j;i++) 67 { 68 while(i<j&&dis[i]+dis[j]>md) j--; 69 if(i>=j) break; 70 res+=j-i; 71 } 72 return res; 73 } 74 75 inline int getans(int u,int fa) 76 { 77 getsize(u,fa); 78 mn=INF; 79 getroot(fa,u,fa); 80 int rt=root; 81 int res=calc(rt,rt,0); 82 vis[root]=true; 83 for(int i=head[rt];~i;i=next[i]) 84 if(to[i]!=fa&&!vis[to[i]]) 85 { 86 res-=calc(to[i],rt,len[i]); 87 res+=getans(to[i],rt); 88 } 89 return res; 90 } 91 92 inline void go() 93 { 94 memset(vis,0,sizeof vis); 95 printf("%d\n",getans(1,1)); 96 } 97 98 int main() 99 { 100 while(scanf("%d%d",&n,&md),n||md) read(),go(); 101 return 0; 102 }