点分治

戳这里

bzoj1468

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define Maxn 80010
using namespace std;

struct edge{
    int to,v,next;
}p[Maxn];
int tot,root,sum,ans,k;
int head[Maxn],vis[Maxn],son[Maxn],bal[Maxn],st[Maxn],dist[Maxn];
void addedge(int a,int b,int c){
    p[tot].to=b;
    p[tot].v=c;
    p[tot].next=head[a];
    head[a]=tot++;
}
void getroot(int u,int fa){
    son[u]=1,bal[u]=0;
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(fa==v||vis[v]) continue;
        son[u]+=son[v];
        bal[u]=max(bal[u],son[v]);
    }
    bal[u]=max(bal[u],sum-son[u]);
    if(bal[u]<bal[root]) root=u;
}
void dfs(int u,int fa){
    st[++st[0]]=dist[u];
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(v==fa||vis[v]) continue;
        dist[v]=dist[u]+p[i].v;
        dfs(v,u);
    }
}
int cal(int u,int val){
    dist[u]=val,st[0]=0;
    dfs(u,-1);
    sort(st+1,st+1+st[0]);
    int res=0;
    for(int i=1,j=st[0];i<j;){
        if(st[i]+st[j]<=k){res+=j-i;i++;}
        else j--;
    }
    return res;
}
void work(int u){
    ans+=cal(u,0);
    vis[u]=1;
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(vis[v]) continue;
        ans-=cal(v,p[i].v);
        sum=son[v],root=0;
        getroot(v,-1);
        work(root);
    }
}
int main()
{
    int n,a,b,c;
    cin>>n;
    memset(head,-1,sizeof head);
    tot=0;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&a,&b,&c);
        addedge(a,b,c);
        addedge(b,a,c);
    }
    cin>>k;
    ans=0;
    sum=n,bal[root=0]=1<<30;
    getroot(1,-1);
    work(root);
    cout<<ans<<endl;
    return 0;
}

bzoj2152

#include<iostream>
#include<cstdio>
#include<cstring>
#define Maxn 40010
using namespace std;

struct edge{
    int to,v,next;
}p[Maxn];
int head[Maxn],vis[Maxn],bal[Maxn],mod[Maxn],dist[Maxn],son[Maxn];
int tot,sum,root,ans;
const int inf=0x3f3f3f3f;
void addedge(int a,int b,int c){
    p[tot].to=b;
    p[tot].v=c;
    p[tot].next=head[a];
    head[a]=tot++;
}
void getroot(int u,int fa){ //sum,root
    son[u]=1,bal[u]=0;
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(v==fa||vis[v]) continue;
        getroot(v,u);
        son[u]+=son[v];
        bal[u]=max(bal[u],son[v]);
    }
    bal[u]=max(bal[u],sum-son[u]);
    if(bal[u]<bal[root]) root=u;
}
void dfs(int u,int fa){
    mod[dist[u]]++;
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(v==fa||vis[v]) continue;
        dist[v]=(dist[u]+p[i].v)%3;
        dfs(v,u);
    }
}
int cal(int u,int val){
    mod[0]=mod[1]=mod[2]=0;
    dist[u]=val%3;
    dfs(u,-1);
    return mod[0]*mod[0]+mod[1]*mod[2]*2;
}
void work(int u){
    ans+=cal(u,0);
    vis[u]=1;
    for(int i=head[u];i!=-1;i=p[i].next){
        int v=p[i].to;
        if(vis[v]) continue;
        ans-=cal(v,p[i].v);
        sum=son[v],root=0;
        getroot(v,-1);
        work(root);
    }
}
int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}
int main()
{
    int n,a,b,c;
    cin>>n;
    tot=ans=0;
    memset(head,-1,sizeof head);
    memset(vis,0,sizeof vis);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&a,&b,&c);
        addedge(a,b,c);
        addedge(b,a,c);
    }
    sum=n,bal[root=0]=inf;
    getroot(1,-1);
    work(root);
    int t=gcd(n*n,ans);
    printf("%d/%d\n",ans/t,n*n/t);
    return 0;
}