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);}