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