【poj1741 Tree】

 1 #include<iostream>
 2 #include<cstdlib>
 3 #include<cstdio>
 4 #include<cstring>
 5 #include<algorithm>
 6 using namespace std;
 7 struct nd{
 8     int to,next,v;
 9 }    e[20005];
10 int head[10005],dep[10005],d[10005],mx[10005],sz[10005];
11 bool vis[10005];
12 int sum,ans,rt;
13 int n,k,cnt;
14 inline void insert(int u,int v,int w){
15     e[++cnt].next=head[u];
16     head[u]=cnt;
17     e[cnt].to=v;
18     e[cnt].v=w;
19 }
20 inline void findG(int now,int fa){
21     mx[now]=0;
22     for(int i=head[now];i;i=e[i].next){
23         if(e[i].to==fa || vis[e[i].to])    continue;
24         findG(e[i].to,now);
25         mx[now]=max(mx[now],sz[e[i].to]);
26     }
27     mx[now]=max(mx[now],sum-sz[now]);
28     if(mx[now]<mx[rt])    rt=now;
29 }
30 inline void findS(int now,int fa){
31     sz[now]=1;
32     for(int i=head[now];i;i=e[i].next){
33         if(vis[e[i].to] || e[i].to==fa)    continue;
34         findS(e[i].to,now);
35         sz[now]+=sz[e[i].to];
36     }
37 }
38 inline void findD(int now,int fa){
39     dep[++dep[0]]=d[now];
40     for(int i=head[now];i;i=e[i].next){
41         if(vis[e[i].to] || e[i].to==fa)    continue;
42         d[e[i].to]=d[now]+e[i].v;
43         findD(e[i].to,now);
44     }
45 }
46 inline int cal(int now,int x){
47     d[now]=x;dep[0]=0;
48     findD(now,-1);
49     sort(dep+1,dep+dep[0]+1);
50     int num=0,l,r;
51     for(int l=1,r=dep[0];l<r;){
52         if(dep[l]+dep[r]<=k){
53             num+=r-l;l++;
54         }
55         else    r--;
56     }
57     return num;
58 }
59 inline void work(int x)
60 {
61     ans+=cal(x,0);
62     vis[x]=1;
63     for(int i=head[x];i;i=e[i].next){
64         if(vis[e[i].to])    continue;
65         ans-=cal(e[i].to,e[i].v);
66         findS(e[i].to,x);
67         sum=sz[e[i].to];
68         rt=0;
69         findG(e[i].to,x);
70         work(rt);
71     }
72 }
73 int main(){
74     int u,v,w;
75     while(1){
76         ans=0,cnt=0;
77         memset(vis,0,sizeof(vis));
78         memset(head,0,sizeof(head));
79         scanf("%d%d",&n,&k);
80         if(!n)    return 0;
81         for(int i=1;i<n;i++){
82             scanf("%d%d%d",&u,&v,&w);
83             insert(u,v,w);
84             insert(v,u,w);
85         }
86         rt=0;
87         mx[rt]=1e9+7;
88         findS(1,-1);
89         sum=sz[1];
90         findG(1,-1);
91         work(rt);
92         printf("%d\n",ans);
93     }
94 }