POJ1741 Tree

Description

给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K

Input

N（n<=40000） 接下来n-1行边描述管道，按照题目中写的输入 接下来是k

Output

一行，有多少对点之间的距离小于等于k

Sample Input

7
1 6 13
6 3 9
3 5 7
4 1 3
2 4 20
4 7 2
10

Sample Output

5

题解：
看了hzwer的代码感觉豁然开朗，这题的点分居然这么简单，原来一直以为相当难。。。
基本思路就是
找重心+统计+去重+点分治
算法的实现有很多细节，hzwer的代码十分优美。学习了。
代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 40000+1000

#define maxm 500+100

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 1000000007

using namespace std;

inline int read()

{

    int x=0,f=1;char ch=getchar();

    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}

    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}

    return x*f;

}
struct edge{int go,next,w;}e[2*maxn];
int ans,tot,sum,n,k,root,head[maxn],s[maxn],f[maxn],deep[maxn],d[maxn];
bool vis[maxn];
inline void insert(int x,int y,int z)
{
    e[++tot].go=y;e[tot].next=head[x];head[x]=tot;e[tot].w=z;
    e[++tot].go=x;e[tot].next=head[y];head[y]=tot;e[tot].w=z;
}
inline void getroot(int x,int fa)
{
    s[x]=1;f[x]=0;
    for(int i=head[x],y;i;i=e[i].next)
    {
        if(vis[y=e[i].go]||y==fa)continue;
        getroot(y,x);
        s[x]+=s[y];
        f[x]=max(f[x],s[y]);
    }
    f[x]=max(f[x],sum-s[x]);
    if(f[x]<f[root])root=x;
}
inline void getdeep(int x,int fa)
{
    deep[++deep[0]]=d[x];
    for(int i=head[x],y;i;i=e[i].next)
    {
        if(vis[y=e[i].go]||y==fa)continue;
        d[y]=d[x]+e[i].w;
        getdeep(y,x);
    }
}
inline int calc(int x,int now)
{
    deep[0]=0;d[x]=now;
    getdeep(x,0);
    sort(deep+1,deep+deep[0]+1);
    int t=0;
    for(int l=1,r=deep[0];l<r;)
    if(deep[l]+deep[r]<=k){t+=r-l;l++;}else r--;
    return t;
}
inline void work(int x)
{
    vis[x]=1;
    ans+=calc(x,0);
    for(int i=head[x],y;i;i=e[i].next)
    {
        if(vis[y=e[i].go])continue;
        ans-=calc(y,e[i].w);
        sum=s[y];
        root=0;
        getroot(y,0);
        work(root);
    }
}

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);
    while(1)
    {
     memset(head,0,sizeof(head));tot=ans=0;
     memset(vis,0,sizeof(vis));

     n=read();k=read();if(n==0)break;
     for1(i,n-1){int x=read(),y=read(),z=read();insert(x,y,z);};
     sum=n;f[root=0]=inf;
     getroot(1,0);
      work(root);
     printf("%d\n",ans);
    }

     return 0;

}