poj1741 树上的点分治

题意：

一棵10000个点的树，每条边的长不超过1000，给定一个值k，问距离不超过k的点对数有多少。（多组数据）

输入样例：

5 4
1 2 3
1 3 1
1 4 2
3 5 1
0 0

输出样例：
8

————————————————————————————————————————————————————————————————————————————
这个题目是在CODEVS 2756 树上的路径 的学习过程中看到别人的题解的过程中看到的，这个题相对简单，是做那个题的基础。基本上是人家的代码，和抄也差不了多少。是学习树上的点分治的一个基础题。
根据一个大神漆子超的论文，树上的分治可以分为点分治，边分治和链分治。点分治最差log,边分治最差n。而链分治是最好的，不过没有学会，有空再从网上找来学习一下，就是只学习点皮毛，对自己做题思路也是一种扩展。
点分治，就是在树中去掉一个点，从而使原来的树变成更小的树，从而达到分治的目的。为了防止极差情况的出现，提高分治的效率，最好的分治点是树的重心（去掉该点后，所得最大子树最小）。所在就有了点分治的方法。
1、深搜，得到各个点为根的子树的大小和各个点的最大子树的大小
2、二次深搜，得到去掉各个点得到的最大子树的大小，从而通过遍历得到重心。
3、应用问题与重心的关系进行分治。
（本题中：
3.1、计算通过重心且不超过k的点对数（包含重边）。
3.2、减去过重心且不超过k的点对数（即有重边的点对数）。
3.3、标明已查到的重心（vis），在新生成的子树中重新进行上述操作，完成分治。
）

边分治，没有做过，不过据漆子超说，是去掉一条过，把树分成2个树，完成分治。个人愚见，如果不是星型那种极个别的情况也可以实现log。
————————————————————————————————————————————————————————————————————————————

 #include<cstdio>
 #include<iostream>
 #include<cstring>
 #include<algorithm>

 using namespace std;
 const int maxn=;

 int n,k,ans;
 struct edge
 {
     int u,v,w,next;
 }e[*maxn];
 int head[maxn],js,jst,mi,root;
 int siz[maxn],mx[maxn],dis[maxn];
 bool vis[maxn];
 void init()
 {
     memset(head,,sizeof(head));
     memset(vis,,sizeof(vis));
     js=;
     ans=;
 }
 void readint(int &x)
 {
     int f=;
     char c=getchar();
     for(;c<''||c>'';c=getchar())if(c=='-')f=-f;
     x=;
     for(;c>=''&&c<='';c=getchar())x=x*+c-'';
     x=x*f;
 }
 void addage(int u,int v,int w)
 {
     e[++js].u=u;e[js].v=v;e[js].w=w;
     e[js].next=head[u];head[u]=js;
 }
 void dfssize(int u,int fa)
 {
     siz[u]=;mx[u]=;
     for(int i=head[u];i;i=e[i].next)
     {
         int v=e[i].v;
         if(v!=fa && !vis[v])
         {
             dfssize(v,u);
             siz[u]+=siz[v];
             if(siz[v]>mx[u])mx[u]=siz[v];
         }
     }
 }
 void dfsroot(int r,int u,int f)
 {
     if(siz[r]-siz[u]>mx[u])mx[u]=siz[r]-siz[u];
     if(mx[u]<mi)
     {
         mi=mx[u];
         root=u;
     }
     for(int i=head[u];i;i=e[i].next)
     {
         int v=e[i].v;
         if(v!=f &&!vis[v])
         {
             dfsroot(r,v,u);
         }
     }
 }
 void dfsdis(int u,int d,int f)
 {
     dis[jst++]=d;
     for(int i=head[u];i;i=e[i].next)
     {
         int v=e[i].v;
         if(v!=f && !vis[v])
         dfsdis(v,d+e[i].w,u);
     }
 }
 int calc(int u,int d)
 {
     int anst=;
     jst=;
     dfsdis(u,d,);
     sort(dis,dis+jst);
     int i=,j=jst-;
     while(i<j)
     {
         while(dis[i]+dis[j]>k && i<j)j--;
         anst+=j-i;
         i++;
     }
     return anst;
 }
 void dfs(int u)
 {
     mi=n;
     dfssize(u,);
     dfsroot(u,u,);
     ans+=calc(root,);
     vis[root]=;
     for(int i=head[root];i;i=e[i].next)
     {
         int v=e[i].v;
         if(!vis[v])
         {
             ans-=calc(v,e[i].w);
             dfs(v);
         }

     }

 }
 int main()
 {
     while(scanf("%d%d",&n,&k)==)
     {
         if(!n && !k)break;
         init();
         for(int u,v,w,i=;i<n;i++)
         {
             readint(u);readint(v);readint(w);
             addage(u,v,w);addage(v,u,w);
         }
         dfs();
         printf("%d\n",ans);
     }
     return ;
 }