BZOJ 3365 [Usaco2004 Feb]Distance Statistics 路程统计 树的点分治

题目大意：给出n(n<=50000)个结点的树，求有多少条路径长度不超过K

求解有关树上的路径问题可以使用树的点分治。

算法流程：
1.选取当前子树的重心，将无根树转为有根树
2.处理过当前根的路径对答案的贡献
3.标记根节点，表示删除根节点
4.递归处理每一个子树

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 50004
using namespace std;
struct Edge {
    int from,to,nxt,val;
    Edge() {}
    Edge(int _from,int _to,int _nxt,int _val):
        from(_from),to(_to),nxt(_nxt),val(_val) {}
}e[N*2];
int n,m,K,tot=-1,fir[N];
void Add_Edge(int x,int y,int v) {
    e[++tot]=Edge(x,y,fir[x],v), fir[x]=tot;
    e[++tot]=Edge(y,x,fir[y],v), fir[y]=tot;
    return ;
}
int root,sum,top,pa[N],siz[N],max_siz[N],s[N],dpt[N];
bool vis[N];
void dfs_root(int x,int from) {
    siz[x]=1; max_siz[x]=0; pa[x]=from;
    for(int i=fir[x];~i;i=e[i].nxt) {
        if(e[i].to==from || vis[e[i].to]) continue;
        dfs_root(e[i].to,x);
        siz[x]+=siz[e[i].to];
        max_siz[x]=max(max_siz[x],siz[e[i].to]);
    }
    max_siz[x]=max(max_siz[x],sum-siz[x]);
    if(max_siz[x]<max_siz[root]) root=x;
    return ;
}
void get_root(int x) {
    sum=siz[x];
    root=0;
    dfs_root(x,0);
    siz[pa[root]]=sum-siz[root];
    return ;
}
void dfs_dpt(int x,int from) {
    s[++top]=dpt[x];
    for(int i=fir[x];~i;i=e[i].nxt) {
        if(e[i].to==from || vis[e[i].to]) continue;
        dpt[e[i].to]=dpt[x]+e[i].val;
        dfs_dpt(e[i].to,x);
    }
    return ;
}
int ans;
int calc(int x,int v) {
    dpt[x]=v;
    top=0;
    dfs_dpt(x,0);
    sort(s,s+top+1);
    int l=1,r=top,sum=0;
    while(l<r) {
        if(s[l]+s[r]<=K) sum+=r-l, l++;
        else r--;
    }
    return sum;
}
void Divide(int x) {
    ans+=calc(x,0);
    vis[x]=true;
    for(int i=fir[x];~i;i=e[i].nxt) {
        if(vis[e[i].to]) continue;
        ans-=calc(e[i].to,e[i].val);
        get_root(e[i].to);
        Divide(root);
    }
    return ;
}
int main() {
    memset(fir,-1,sizeof fir);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++) {
        int x,y,v;
        char dir[5];
        scanf("%d%d%d%s",&x,&y,&v,dir);
        Add_Edge(x,y,v);
    }
    scanf("%d",&K);
    max_siz[0]=siz[1]=n;
    get_root(1);
    Divide(root);
    printf("%d\n",ans);
    return 0;
}