bzoj 3365（点分治）

传送门
题解：裸的点分治，统计距离不超过k的点对数量。好像题目没说是一棵树，但是应该是默认的，不然两点的距离就不唯一了。
注意：点分治中getdis( )要再求一次size，以前本人写bzoj 1468的题解时扯的那些*话就当没听见，什么整体偏大不影响都是扯淡的，这个操作不执行就会导致重心找偏，有些题数据水一样能A，但是另外的就会出问题。
这次真的是稳稳地自己调出来的。<(￣︶￣)>

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int MAXN=4e4+8;int n,m,K,siz[MAXN],mx[MAXN],sum,root,ans=0,tot;ll a[MAXN],dis[MAXN];bool vis[MAXN];int head[MAXN],edge=0;struct EDGE {    int v,nxt,w;}e[MAXN<<1];void getroot(int p,int fa) {    siz[p]=1,mx[p]=0;    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (v^fa&&!vis[v]) {            getroot(v,p);            siz[p]+=siz[v];            mx[p]=max(mx[p],siz[v]);         }    }    mx[p]=max(mx[p],sum-siz[p]);    if (mx[p]<mx[root]) root=p;}void getdis(int p,int fa) {    siz[p]=1,a[++tot]=dis[p];    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (v^fa&&!vis[v]) {            dis[v]=dis[p]+e[i].w;            getdis(v,p);            siz[p]+=siz[v];//!!!        }    }}inline int cal(int p,int val) {    int ret=0;    tot=0;    dis[p]=val;    getdis(p,0);    sort(a+1,a+tot+1);    for (int l=1,r=tot;l<r;++l) {        while (a[r]+a[l]>K&&l<r)            --r;        ret+=r-l;    }    return ret;}void work(int p) {    ans+=cal(p,0);    vis[p]=true;    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (!vis[v]) {            ans-=cal(v,e[i].w);            sum=siz[v],root=0;            getroot(v,0);            work(root);        }    }}inline int read() {    int x=0,f=1;char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();    return x*f;}inline void adde(int u,int v,int w) {    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;}int main() {//  freopen("bzoj 3365.in","r",stdin);    memset(head,-1,sizeof(head));    memset(vis,false,sizeof(vis));    n=read(),m=read();    for (register int i=1;i<=m;++i) {        int u=read(),v=read(),w=read();        char st[2];scanf("%s",st);        adde(u,v,w);    }    K=read();    sum=n,root=0,mx[0]=MAXN;    getroot(1,0);    work(root);    printf("%d\n",ans);    return 0;}