POJ 1741 Tree ——（树分治）

　　思路参考于：http://blog.csdn.net/yang_7_46/article/details/9966455，不再赘述。

　　复杂度：找树的重心然后分治复杂度为logn，每次对距离数组dep排序复杂度为nlogn，而找重心的复杂度为dfs的复杂度——O(n)，因此总的复杂度为O(nlognlogn)。

　　因为第一次写树分治，留个代码：

 #include <stdio.h>
 #include <algorithm>
 #include <string.h>
 #include <vector>
 using namespace std;
 const int N =  + ;

 struct edge
 {
     int v,w;
 };
 int n,k,root,sz,ans;
 bool vis[N];
 int son[N],f[N],d[N];
 vector<edge> G[N];
 vector<int> dep;
 void init()
 {
     memset(vis,false,sizeof(vis));
     for(int i=;i<=n;i++) G[i].clear();
     memset(son,,sizeof son);
     memset(f,,sizeof f);
 }
 void  addEdge(int u,int v,int w)
 {
     G[u].push_back((edge){v,w});
     G[v].push_back((edge){u,w});
 }
 void getRoot(int u,int fa)
 {
     son[u] = ; f[u] = ;
     for(int i=;i<G[u].size();i++)
     {
         edge e = G[u][i];
         int v = e.v;
         if(v == fa || vis[v]) continue;
         getRoot(v,u);
         son[u] += son[v];
         f[u] = max(f[u], son[v]);
     }
     f[u] = max(f[u], sz-son[u]);
     if(f[u] < f[root]) root = u;
 }
 void getDep(int u,int fa)
 {
     dep.push_back(d[u]);
     for(int i=;i<G[u].size();i++)
     {
         edge e = G[u][i];
         int v = e.v, w= e.w;
         if(v == fa || vis[v]) continue;
         d[v] = d[u] + w;
         getDep(v,u);
     }
 }
 int cal(int u,int len)
 {
     dep.clear(); d[u] = len;
     getDep(u, );
     sort(dep.begin(), dep.end());
     int ret = ;
     for(int l=,r=dep.size()-;l<r;)
     {
         if(dep[l] + dep[r] <= k) ret += r - (l++);
         else r--;
     }
     return ret;
 }
 void solve(int u)
 {
     ans += cal(u, );
     vis[u] = ;
     for(int i=;i<G[u].size();i++)
     {
         edge e = G[u][i];
         int v = e.v, w = e.w;
         if(vis[v]) continue;
         ans -= cal(v, w);
         f[] = sz = son[v];
         getRoot(v, root=);
         solve(root);
     }
 }

 int main()
 {
     while(scanf("%d%d",&n,&k) == )
     {
         if(n ==  && k == ) break;
         init();
         for(int i=;i<=n-;i++)
         {
             int u,v,w;
             scanf("%d%d%d",&u,&v,&w);
             addEdge(u,v,w);
         }
         f[] = sz = n;
         getRoot(,root=);
         ans = ;
         solve(root);
         printf("%d\n",ans);
     }
     return ;
 }