BZOJ 1468: Tree 点分治题解

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 1604 Solved: 874

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男人八题系列

---

貌似是楼天成的题？？？

---

点分的时候two-pointer解决，然后突然发现一个很玄学的东西，就是以前每次在cal的时候都没有重新找sz，其实以前的sz都是错的，也就是说找的可能不是我们的重心，好玄学啊，然而还是过了，说明随机化的话还是可以的，虽然也不是很随机

---

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int MAXN=40000+10;
struct Line{
    int to,nxt,flow;
}line[MAXN*2];
int ans,N,K,a[MAXN],dis[MAXN],sz[MAXN],sum,vis[MAXN],root,mx[MAXN],head[MAXN],tail;
void add_line(int from,int to,int flow){line[++tail].to=to;line[tail].nxt=head[from];head[from]=tail;line[tail].flow=flow;}
void getroot(int u,int fa){
    sz[u]=1;mx[u]=0;
    for(register int i=head[u];i;i=line[i].nxt){
        int v=line[i].to;
        if(vis[v]||v==fa) continue;
        getroot(v,u);
        sz[u]+=sz[v];
        mx[u]=max(mx[u],sz[v]);
    }
    mx[u]=max(mx[u],sum-mx[u]);
    if(mx[u]<mx[root])root=u;
}
void getdeep(int u,int fa){
    sz[u]=1;a[++tail]=dis[u];
    for(register int i=head[u];i;i=line[i].nxt){
        int v=line[i].to;
        if(v==fa||vis[v]) continue;
        dis[v]=dis[u]+line[i].flow;
        getdeep(v,u);
        sz[u]+=sz[v];
    }
}
int cal(int u,int d){
    dis[u]=d;tail=0;
    getdeep(u,0);
    sort(a+1,a+tail+1);
    int l=1,r=tail,sum=0;
    while(l<r){
        if(a[l]+a[r]<=K)sum+=r-l,l++;
        else                     r--;
    } 
    return sum;
}
void solve(int u){
    vis[u]=1;
    ans+=cal(u,0);
    for(register int i=head[u];i;i=line[i].nxt){
        int v=line[i].to;
        if(vis[v]) continue;
        ans-=cal(v,line[i].flow);
        root=0;sum=sz[v];
        getroot(v,0);
        solve(root);
    }
}
int main(){
    scanf("%d",&N);
    for(register int i=1;i<=N-1;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);add_line(u,v,w);add_line(v,u,w);}
    scanf("%d",&K);
    mx[0]=0x7ffffff;root=0;sum=N;
    getroot(1,0);
    solve(root);
    printf("%d\n",ans);
}