1741 Tree
【题目描述】
传送门
【题解】
点分治模板。
代码如下
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 10005
#define LL long long
using namespace std;
int n,K,Rot,RotSize;
int Siz[MAXN],dst[MAXN],Maxs[MAXN];LL Ans;
vector<int> Now;
bool vis[MAXN];
struct Edge{
int tot,lnk[MAXN],son[MAXN<<1],nxt[MAXN<<1],W[MAXN<<1];
void clean(){memset(lnk,0,sizeof(lnk));tot=0;}
void Add(int x,int y,int z){son[++tot]=y;W[tot]=z;nxt[tot]=lnk[x];lnk[x]=tot;}
}E;
void Get_Rot(int x,int fa){
Siz[x]=1;Maxs[x]=0;
for(int j=E.lnk[x];j;j=E.nxt[j])
if(!vis[E.son[j]]&&E.son[j]!=fa){
Get_Rot(E.son[j],x);
Siz[x]+=Siz[E.son[j]];
Maxs[x]=max(Siz[E.son[j]],Maxs[x]);
}
Maxs[x]=max(Maxs[x],RotSize-Siz[x]);
if(Maxs[x]<Maxs[Rot]) Rot=x;
}
void Get_Dst(int x,int f){
Siz[x]=1;Now.push_back(dst[x]);
for(int j=E.lnk[x];j;j=E.nxt[j])
if(E.son[j]!=f&&!vis[E.son[j]]){
dst[E.son[j]]=dst[x]+E.W[j];
Get_Dst(E.son[j],x);Siz[x]+=Siz[E.son[j]];
}
}
int Cal(int x,int y){
int Ret=0;
Now.clear();dst[x]=y;Get_Dst(x,0);
sort(Now.begin(),Now.end());
for(int l=0,r=Now.size()-1;l<r;l++){
while(Now[l]+Now[r]>K&&l<r) r--;
Ret+=r-l;
}
return Ret;
}
void Solve(int x){
Ans+=Cal(x,0);vis[x]=1;
for(int j=E.lnk[x];j;j=E.nxt[j])
if(!vis[E.son[j]]){
Ans-=Cal(E.son[j],E.W[j]);
Maxs[0]=RotSize=Siz[E.son[j]];Rot=0;
Get_Rot(E.son[j],0);Solve(Rot);
}
}
int main(){
#ifndef ONLINE_JUDGE
freopen("POJ1741.in","r",stdin);
freopen("POJ1741.out","w",stdout);
#endif
while(scanf("%d%d",&n,&K),n||K){
memset(vis,0,sizeof(vis));
E.clean();
for(int i=1;i<n;i++){
int x,y,z;scanf("%d%d%d",&x,&y,&z);
E.Add(x,y,z);E.Add(y,x,z);
}
Ans=0;Maxs[0]=RotSize=n;Rot=0;
Get_Rot(1,0);Solve(Rot);
printf("%lld\n",Ans);
}
return 0;
}