【poj1741】tree 点分治

【题目大意】

给一颗n个节点的树，每条边上有一个距离v（v<=1000）。

定义d(u,v)为u到v的最小距离；

给定k值，求有多少点对（u，v）使u到v的距离小于等于k。

其中n<=10000,k<2^31

【题解】

蒟蒻发现很久以前学的点分治已经忘得差不多了，于是恶补一发

所谓点分治，就是枚举每一个点为根的情况，计算当前根对答案的贡献。

具体可以参考http://www.cnblogs.com/chty/p/5912360.html（这是我以前的博客）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 10010
#define INF 1000000000
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i)
namespace INIT{
char buf[1<<15],*fs,*ft;
inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
inline int read(){
int x=0,f=1; char ch=getc();
while(!isdigit(ch)) {if(ch=='-') f=-1; ch=getc();}
while(isdigit(ch)) {x=x*10+ch-'0'; ch=getc();}
return x*f;
}
}using namespace INIT;
struct node{int y,next,v;}e[MAXN*2];
int n,k,root,sum,len,ans,Link[MAXN],f[MAXN],dis[MAXN],deep[MAXN],vis[MAXN],size[MAXN];
void insert(int x,int y,int v){e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;}
void getroot(int x,int fa){
size[x]=1; f[x]=0;
for(int i=Link[x];i;i=e[i].next){
if(e[i].y==fa||vis[e[i].y]) continue;
getroot(e[i].y,x); size[x]+=size[e[i].y];
f[x]=max(f[x],size[e[i].y]);
}
f[x]=max(f[x],sum-size[x]);
if(f[x]<f[root]) root=x;
}
void getdeep(int x,int fa){
deep[++deep[0]]=dis[x];
for(int i=Link[x];i;i=e[i].next){
if(e[i].y==fa||vis[e[i].y]) continue;
dis[e[i].y]=dis[x]+e[i].v; getdeep(e[i].y,x);
}
}
int cal(int x,int v){
dis[x]=v; deep[0]=0; getdeep(x,0);
sort(deep+1,deep+deep[0]+1);
int l=1,r=deep[0],temp(0);
while(l<r) {if(deep[l]+deep[r]<=k)temp+=r-l,++l;else --r;}
return temp;
}
void solve(int x){
ans+=cal(x,0); vis[x]=1;
for(int i=Link[x];i;i=e[i].next){
if(vis[e[i].y]) continue;
ans-=cal(e[i].y,e[i].v);
root=0; sum=size[e[i].y];
getroot(e[i].y,0);solve(root);
}
}
void pre(){
memset(Link,0,sizeof(Link));
memset(vis,0,sizeof(vis));
ans=root=len=0;
}
int main(){
freopen(FILE".in","r",stdin);
freopen(FILE".out","w",stdout);
while(1){
n=sum=read();k=read();if(!n&&!k)break; pre();
up(i,1,n-1){int x=read(),y=read(),v=read();insert(x,y,v);insert(y,x,v);}
f[0]=INF; getroot(1,0); solve(root);
printf("%d\n",ans);
}
return 0;
}