树分治（Tree，poj 1741）

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0X3F3F3F3F
#define maxn 10010

int n,k;
int tot;
bool vis[maxn];
int dp[maxn];
int temp[maxn];
int MIN;
int le,ri;

struct Edge
{
int to,w,next;
}edges[maxn<<1];
int head[maxn];

void init()
{
tot=0;
memset(head,-1,sizeof(head));
}

void addedge(int u,int v,int w)
{
edges[tot].to=v;
edges[tot].w=w;
edges[tot].next=head[u];
head[u]=tot++;
}

int dfs(int u,int f)
{
dp[u]=1;
for(int i=head[u];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
if(vis[v]||v==f) continue;
dp[u]+=dfs(v,u);
}
return dp[u];
}

void get(int u,int f,int tot,int& root)
{
int MAX=tot-dp[u];
for(int i=head[u];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
if(vis[v]||v==f) continue;
get(v,u,tot,root);
MAX=max(MAX,dp[v]);
}
if(MAX<MIN)
{
MIN=MAX;
root=u;
}
}

void add(int u,int f,int sum)
{
temp[ri++]=sum;
for(int i=head[u];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
if(vis[v]||v==f) continue;
add(v,u,sum+edges[i].w);
}
}

int cul(int l,int r)
{
sort(temp+l,temp+r);
int ret=0;
int e=r-1;
for(int i=l;i<r;i++)
{
if(temp[i]>k) break;
while(e>=l&&temp[i]+temp[e]>k) e--;
ret+=e-l+1;
if(e>=i) ret--;
}
return ret>>1;
}

int solve(int u)
{
int tot=dfs(u,-1);
int root;
MIN=INF;
get(u,-1,tot,root);
int ret=0;
vis[root]=true;
for(int i=head[root];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
if(vis[v]||v==root) continue;
ret+=solve(v);
}
le=ri=0;
for(int i=head[root];i!=-1;i=edges[i].next)
{
int v=edges[i].to;
if(vis[v]||v==root) continue;
add(v,root,edges[i].w);
ret-=cul(le,ri);
le=ri;
}
ret+=cul(0,ri);
for(int i=0;i<ri;i++)
if(temp[i]<=k) ret++;
else break;
vis[root]=false;
return ret;
}

int main()
{
while(scanf("%d %d",&n,&k)==2&&(n+k))
{
init();
int u,v,w;
for(int i=1;i<n;i++)
{
scanf("%d %d %d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
printf("%d\n",solve(1));
}
return 0;
}