【POJ1741】Tree（点分治）

题面

Vjudge
题目大意：
求树中距离小于$K$的点对的数量

题解

完全不觉得点分治了。。
简直$GG$，更别说动态点分治了。。。

于是来复习一下。

对于每一层分治重心
求出它到子树中任意点的距离
然后$two-pointers$计算满足大于 K <script type="math/tex" id="MathJax-Element-108">K</script>的点对的数目，累加到答案中，
但是同一棵子树内的会算重
所以再对于每一棵子树，减去字数内两两满足要求的点对
完全不记得了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 11111
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next,w;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int root,Size,rmx;
int S[MAX],dep[MAX],tot;
int size[MAX],K,n,ans;
bool vis[MAX];
void Getroot(int u,int ff)
{
    size[u]=1;int mx=0;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==ff||vis[v])continue;
        Getroot(v,u);
        size[u]+=size[v];mx=max(mx,size[v]);
    }
    mx=max(mx,Size-size[u]);
    if(mx<rmx)rmx=mx,root=u;
}
void getdep(int x,int ff)
{
    S[++tot]=dep[x];
    for(int i=h[x];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==ff||vis[v])continue;
        dep[v]=dep[x]+e[i].w;
        getdep(v,x);
    }
}
int Calc(int u,int pre)
{
    tot=0;dep[u]=pre;
    getdep(u,0);
    int l=1,r=tot,ret=0;
    sort(&S[1],&S[tot+1]);
    while(l<r)
    {
        if(S[l]+S[r]<=K)ret+=r-l,++l;
        else --r;
    }
    return ret;
}
void Solve(int u)
{
    ans+=Calc(u,0);vis[u]=true;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(vis[v])continue;
        ans-=Calc(v,e[i].w);
        Size=rmx=size[v];
        Getroot(v,u);
        Solve(root);
    }
}
int main()
{
    while(233)
    {
        rmx=Size=n=read();K=read();
        if(!n&&!K)break;
        memset(vis,0,sizeof(vis));ans=0;
        memset(h,0,sizeof(h));cnt=1;
        for(int i=1;i<n;++i)
        {
            int u=read(),v=read(),w=read();
            Add(u,v,w);Add(v,u,w);
        }
        Getroot(1,0);
        Solve(root);
        printf("%d\n",ans);
    }
}