楼天成男人8题（树的分治-POJ1741）

题目：给定一棵N(1<=N<=10000)个结点的带权树，定义dist(u,v)为u,v两点间的最短路径长度，路径的长度定义为路径上所有边的权和。再给定一个K(1<=K<=10^9)，如果对于不同的两个结点a,b，如果满足dist(a,b)<=K，则称(a,b)为合法点对。求合法点对个数。

算法：树上分治基础题

第一步：枚举中间节点（x是头结点也按照中间节点算）：假设一部分合法点对会经过点x（当然x可以是N个节点中的任意一个）。当然这个x在树形结构中可以按根节点树形递归枚举。

第二步：计算以x为根节点的子树中所有节点到x节点的路径长度放在dep[top]数组中，具体方法是DFS搜索x的所有子树

第三步：计算点对：这一步最为关键 首先将dep[]数组从小到大排序，O（n）复杂度求出dep[]中任意两个和<=K的数量（这里请仔细想想是如何做到的），我们记作sum1。这样算下来会有不合法的点对，因为在同一棵子树中的两个dep[i]+dep[j]<=K 会共用公共边，即有重边。我们在递归的过程中把x为根的子树中会在sum1中重复计算的点对个数都统计在sum2中 最后sum1-sum2即为以x节点为中间节点的点对的个数。依次递归即可得到答案。

#include 
#include 
#include 
#include 
#include 
using namespace std;
#define N 11000
#define inf 0x0f0f0f0f
struct Edge{
    int nxt,val;
    Edge(){}
    Edge(int _n,int _v){
        nxt = _n; val = _v;
    }
};
vector edg[N];
int siz[N],f[N],vis[N],dep[N];
int n,ans,k,root;

int max(int a,int b){ return a>b?a:b; }
int min(int a,int b){ return af[cur]) root = cur;
}
void getDep(int cur,int fa,int len,int &top){
    dep[++top] = len;
    for(int i = 0;i<(int)edg[cur].size();i++){
        int nxt = edg[cur][i].nxt;
        int val = edg[cur][i].val;
        if(vis[nxt]||nxt==fa) continue;
        getDep(nxt,cur,len+val,top);
    }
}
int getTimes(int t_root,int len){
    int top = 0,tmp = 0;
    getDep(t_root,-1,len,top);
    sort(dep+1,dep+top+1);
    for(int l = 1,r = top;l