poj 1741 Tree 点分治

题目大意：
给定一棵N个节点、边上带权的树，再给出一个K，询问有多少个点对(u,v)满足dis[u,v]<=K。
数据规模：

多组测试数据，N≤10000，1≤边权≤1000，1≤K≤10^9。

马克：楼天城男人必做8题之一.... 目前只做了这一题

http://wenku.baidu.com/view/60c6aa1ffc4ffe473368aba8.html

上面网址是讲解思路的论文，具体实现没说。。

首先是要找到重心，那么先递归求出每个点拆掉后能够得到的最大的点的个数，这个点的个数最小的肯定是重心。

然后就按照论文讲解的方法求点对数量，先要求出每个点到重心的距离。

下面是代码：

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <map>
#include <vector>
using namespace std;

const int MAXN = 10100;
const int inf = 0x7f7f7f7f;
typedef pair<int,int> PII;
int n,k;
int vis[MAXN],fa[MAXN];
int mson[MAXN],son[MAXN];
int nod[MAXN],ns;
int dis[MAXN],ds;
int ans;
vector<PII >vv[MAXN];
void dfs(int u,int p)
{
    son[u]=1;
    mson[u]=0;
    nod[ns++]=u;
    int siz=vv[u].size();
    for(int i=0;i<siz;i++)
    {
        int v=vv[u][i].first;
        if(vis[v]||v==p) continue;
        dfs(v,u);
        son[u]+=son[v];
        mson[u]=max(mson[u],son[v]);
    }
}
int getroot(int u)//求重心
{
    ns=0;
    dfs(u,-1);
    int mi=MAXN,res=-1;
    for(int i=0;i<ns;i++)
    {
        mson[nod[i]]=max(mson[nod[i]],son[u]-son[nod[i]]);//每个点拆掉后能够得到的最大的点的个数
        if(mson[nod[i]]<mi)
        {
            mi=mson[nod[i]];
            res=nod[i];
        }
    }
    return res;
}
void getdis(int u,int d,int p)//求出子节点到根的距离
{
    dis[ds++]=d;
    int siz=vv[u].size();
    for(int i=0;i<siz;i++)
    {
        int v=vv[u][i].first;
        if(vis[v]||v==p) continue;
        getdis(v,d+vv[u][i].second,u);
    }
}
void calc(int u,int t)  
{
   sort(dis,dis+ds);
   int i=dis[0]?0:1,j=ds-1;
   while(i<j)
   {
       if(dis[i]+dis[j]<=k)
       {
        ans+=(j-i)*t;
        i++;
       }
       else j--;
   }
}
void solve(int u)
{
    u=getroot(u);
    ds=0;
    getdis(u,0,-1);
    calc(u,1);         //求以u为根的子节点可以构成多少对
    vis[u]=1;
    for(int i=0;i<ds;i++)
        if(dis[i]<=k&&dis[i]!=0) ans++;//以根为起点的个数
    int siz=vv[u].size();
    for(int i=0;i<siz;i++)
    {
        int v=vv[u][i].first;
        if(vis[v]) continue;
        ds=0;
        getdis(v,vv[u][i].second,-1);   
        calc(v,-1);                   //减去属于同一个子树的重复点对
        solve(v);
    }

}
int main()
{
    while(~scanf("%d%d",&n,&k)&&(n||k))
    {
        for(int i=0;i<=n;i++) vv[i].clear();
        for(int i=1;i<n;i++)
        {
            int u,v,l;
            scanf("%d%d%d",&u,&v,&l);
            vv[u].push_back(make_pair(v,l));
            vv[v].push_back(make_pair(u,l));
        }
        ans=0;
        memset(vis,0,sizeof(vis));
        solve(1);
        printf("%d\n",ans);
    }
}