[BZOJ1468]Tree 点分治

点分治模板。
易错点：分治时候传进函数里的size应为(sz[p->t]>sz[rt]?size-sz[rt]:sz[p->t])
代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=40010;
int n,k,top,ans;
bool vis[maxn];
int st[maxn],sz[maxn];
struct edge
{
    int t,w;
    edge *next;
}*con[maxn];
void ins(int x,int y,int w)
{
    edge *p=new edge;
    p->t=y;
    p->w=w;
    p->next=con[x];
    con[x]=p;
}
void getroot(int v,int fa,int size,int &root)
{
    bool flag=1;
    sz[v]=1;
    for(edge *p=con[v];p;p=p->next)
        if(p->t!=fa&&!vis[p->t])
        {
            getroot(p->t,v,size,root);
            sz[v]+=sz[p->t];
            if(sz[p->t]*2>size) flag=0;
        }
    if(size-sz[v]*2>0) flag=0;
    if(flag) root=v;    
}
void dfs(int v,int fa,int d)
{
    st[++top]=d;
    for(edge *p=con[v];p;p=p->next)
        if(!vis[p->t]&&p->t!=fa) dfs(p->t,v,d+p->w);
}
void cal(int v,int f,int d)
{
    top=0;
    dfs(v,-1,d);
    sort(st+1,st+top+1);
    for(int hd=1,tl=top;hd<tl;hd++)
    {
        while(hd<tl&&st[hd]+st[tl]>k) tl--;
        ans+=(tl-hd)*f;
    }
}
void solve(int v,int size)
{
    int rt;
    getroot(v,-1,size,rt);
    vis[rt]=1;
    cal(rt,1,0);
    for(edge *p=con[rt];p;p=p->next)if(!vis[p->t]) cal(p->t,-1,p->w);
    for(edge *p=con[rt];p;p=p->next)if(!vis[p->t]) {solve(p->t,(sz[p->t]>sz[rt]?size-sz[rt]:sz[p->t]));}   
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        ins(x,y,w);
        ins(y,x,w);
    }
    scanf("%d",&k);
    ans=0;
    solve(1,n);
    printf("%d",ans);
    return 0;
}