POJ 1741 树上点分治

题意：

求树上的距离小于等于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 }