【bzoj1468】【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

HINT

Source

LTC男人八题系列

题解：这个题属于点分治中求点对数量的题。

具体思路大致是

1.求出重心

2.统计过重心的答案。

3.减去经过一条边两次的答案。

4.继续分治子树。

这个题统计答案的时候随便用数据结构维护一下或者排个序就好了。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#define N 40010
using namespace std;
struct use{int st,en,v;}e[N*2]; 
int ro,d[N],x,y,v,ans,point[N],n,k,next[N*2],cnt,f[N],size[N],temp,a[N],l,r; 
bool vis[N];
void add(int x,int y,int v){
    next[++cnt]=point[x];point[x]=cnt;
    e[cnt].st=x;e[cnt].en=y;e[cnt].v=v;
}
void dfs(int x,int fa){
    size[x]=1;f[x]=0;
    for (int i=point[x];i;i=next[i])
       if (!vis[e[i].en]&&e[i].en!=fa){
          dfs(e[i].en,x);size[x]+=size[e[i].en];f[x]=max(f[x],size[e[i].en]);   
       } 
    f[x]=max(f[x],temp-size[x]);if (f[x]<f[ro]) ro=x;
}
void pre(int x,int fa){
    a[++a[0]]=d[x];
    for (int i=point[x];i;i=next[i])
        if (!vis[e[i].en]&&e[i].en!=fa){d[e[i].en]=d[x]+e[i].v;pre(e[i].en,x);}
}
int cal(int x,int w){
    int ans(0);
    d[x]=w;a[0]=0;pre(x,0);
    sort(a+1,a+a[0]+1);
    for(l=1,r=a[0];l<r;){if (a[l]+a[r]<=k) {ans+=r-l;l++;}else r--;}
    return ans;
}
void solve(int x){
    ans+=cal(x,0);vis[x]=1;
    for (int i=point[x];i;i=next[i])
        if (!vis[e[i].en]){
           ans-=cal(e[i].en,e[i].v);temp=size[e[i].en];ro=0;
           dfs(e[i].en,0);solve(ro);
        }
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n-1;i++){scanf("%d%d%d",&x,&y,&v);add(x,y,v);add(y,x,v);}
    scanf("%d",&k);f[0]=n+1;
    temp=n;dfs(1,0);solve(ro);
    cout<<ans<<endl; 
}