[bzoj1468][poj1741]Tree[点分治]

可以说是点分治第一题，之前那道的点分治只是模模糊糊，做完这道题感觉清楚了很多，点分治可以理解为每次树的重心（这样会把数分为若干棵子树，子树大小为log级别），然后统计包含重心的整个子树的值减去各个子树的值，这样算出的就是与这个重心有关的情况的答案，比如这道题，求路径，那么就考虑在重心所在的子树中所有的路径减去不过重心的路径就是过重心的路径了。之前重心没找对...poj时间卡的紧就T了。。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>

using namespace std;

struct Edge
{
    int    to,next,w;
}e[210000];

int    n,k,cnt,p[110000],Ans;
int    Son[110000],f[110000],val[110000],depth[110000];
bool    visited[110000];

void    Add_edge(const int x,const int y,const int z)
{
    e[++cnt].to=y;
    e[cnt].next=p[x];
    e[cnt].w=z;
    p[x]=cnt;
    return ;
}

void    Get_root(const int S,const int fa,const int tot,int & root)
{
    Son[S]=1,f[S]=0;
    for(int i=p[S];i;i=e[i].next)
    {
        if(e[i].to==fa || visited[e[i].to])continue;
        Get_root(e[i].to,S,tot,root);
        Son[S]+=Son[e[i].to];
        f[S]=max(f[S],Son[e[i].to]);
    }
    f[S]=max(f[S],tot-Son[S]);
    if(f[S]<f[root])root=S;
    return ;
}

void    Get_depth(const int S,const int fa)
{
    val[++val[0]]=depth[S];
    for(int i=p[S];i;i=e[i].next)
    {
        if(e[i].to==fa || visited[e[i].to])continue;
        depth[e[i].to]=depth[S]+e[i].w;
        Get_depth(e[i].to,S);
    }
    return ;
}

int    Calc(const int S,const int w)
{
    depth[S]=w,val[0]=0;
    Get_depth(S,0);
    sort(val+1,val+val[0]+1);
    int    t=0,l,r;
    for(l=1,r=val[0];l<r;)
    {
        if(val[l]+val[r]<=k)t+=r-l,l++;
        else    r--;
    }
    return t;
}

void    TDC(const int S)
{
    Ans+=Calc(S,0);
    visited[S]=true;
    for(int i=p[S];i;i=e[i].next)
    {
        if(visited[e[i].to])continue;
        Ans-=Calc(e[i].to,e[i].w);
        int root=0;
        Get_root(e[i].to,0,Son[e[i].to],root);
        TDC(root);
    }
    return ;
}

int main()
{
    int    x,y,z,i,root;

    while(scanf("%d%d",&n,&k) && n && k)
    {
        root=0,memset(p,0,sizeof(p));cnt=0;
        memset(visited,0,sizeof(visited));
        Ans=0;
        for(i=1;i<n;++i)
        {
            scanf("%d%d%d",&x,&y,&z);
            Add_edge(x,y,z);
            Add_edge(y,x,z);
        }

        f[0]=0x3f3f3f3f;
        Get_root(1,0,n,root);
        TDC(root);

        printf("%d
",Ans);
    }

    return 0;
}