bzoj1468 Tree（点分治）

点分治，就是每次找到目前子树的重心，把它当做根，处理所有过重心的路径信息，然后递归处理子树。
什么是一棵树的重心呢？选出的一个点，使分出的子树中节点数最大的子树最小，就是重心。这样分出的子树点数不超过n/2+1，就可以保证复杂度了。怎么找重心呢？我们就按定义来找，每次O(n)。然后每次递归时处理过根的合法点对。怎么做呢？我们容斥一下，变成给定一个数组，问有几个有序对满足和＜=K。我们可以排序+线性扫描得出答案。
我的dfs1是处理出sz，dfs2是在子树中找重心。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define N 40010
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,h[N],num=0,sz[N],rt,f[N],sumsz=0,fa[N],a[N],d[N],K,ans=0;
bool vis[N];
struct edge{
    int to,next,val;
}data[N<<1];
void dfs1(int x){
    sz[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        fa[y]=x;dfs1(y);sz[x]+=sz[y];
    }
}
void dfs2(int x){
    f[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||vis[y]) continue;
        dfs2(y);f[x]=max(f[x],sz[y]);
    }f[x]=max(f[x],sumsz-sz[x]);if(f[x]<f[rt]) rt=x;
}
inline void dfs3(int x){
    a[++a[0]]=d[x];
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]||y==fa[x]) continue;
        d[y]=d[x]+data[i].val;dfs3(y);
    }
}
inline int calc(int root,int val){
    d[root]=val;a[0]=0;dfs3(root);sort(a+1,a+a[0]+1);
    int l=1,r=a[0],res=0;
    while(l<r) if(a[l]+a[r]>K) r--;else res+=r-l,l++;return res;
}
inline void solve(int x){
    vis[x]=1;fa[x]=0;dfs1(x);ans+=calc(x,0);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        ans-=calc(y,data[i].val);sumsz=sz[y];rt=0;dfs2(y);solve(rt);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();f[0]=n+1;
    for(int i=1;i<n;++i){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
    }K=read();rt=0;dfs1(1);sumsz=n;dfs2(1);solve(rt);
    printf("%d\n",ans);
    return 0;
}