楼天成男人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 <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
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<Edge> 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 a<b?a:b; }
void getRoot(int cur,int fa,int sum){
    int tmp = 0;  siz[cur] = 1;
    for(int i = 0;i<(int)edg[cur].size();i++){
        int nxt = edg[cur][i].nxt;
        if(vis[nxt]||nxt==fa) continue;
        getRoot(nxt,cur,sum);
        siz[cur] += siz[nxt];
        tmp = max(tmp,siz[nxt]);
    }
    f[cur] = max(tmp,sum-siz[cur]);
    if(f[root]>f[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<r;){
        if(dep[l]+dep[r]<=k){
            tmp += r-l;
            l++;
        }
        else r--;
    }
    return tmp;
}
int getRootID(int cur,int sum){
    root = 0; f[root] = inf;
    getRoot(cur,-1,sum);
    getRoot(root,-1,sum);
    return root;
}
void CDQ(int t_root){
    vis[t_root] = 1;
    ans += getTimes(t_root,0);
    for(int i = 0;i<(int)edg[t_root].size();i++){
        int nxt = edg[t_root][i].nxt;
        int val = edg[t_root][i].val;
        if(vis[nxt]==1) continue;
        ans -= getTimes(nxt,val);
        nxt = getRootID(nxt,siz[nxt]);
        CDQ(nxt);
    }
    vis[t_root] = 0;
}
void solve(int n){
    ans = 0;
    int t_root = getRootID(1,n);
    CDQ(t_root);
    printf("%d\n",ans);
}
void init(){
    for(int i = 1;i<=n;i++) edg[i].clear();
    memset(vis,0,sizeof(vis));
}
int main(){
    int a,b,v;
    //freopen("Test.txt","r",stdin);
    while(~scanf("%d%d",&n,&k)){
        init();
        if(n==0&&k==0) break;
        for(int i = 1;i<=n-1;i++){
            scanf("%d%d%d",&a,&b,&v);
            edg[a].push_back(Edge(b,v));
            edg[b].push_back(Edge(a,v));
        }
        solve(n);
    }
    return 0;
}