Tree POJ - 174

点分模板题

都快改的跟题解一模一样了2333333

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 struct E
 {
     int to,nxt,d;
 }e[];
 int f1[],ne;
 bool vis[];
 int n,k,root,sz[],fx[],sum,ans;
 //sum：当前连通块节点总数
 int dep[],tmp[];
 void getroot(int x,int fa)
 //获取x所在连通块内的根
 {
     sz[x]=;fx[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getroot(e[k].to,x);
             sz[x]+=sz[e[k].to];
             fx[x]=max(fx[x],sz[e[k].to]);
         }
     fx[x]=max(fx[x],sum-sz[x]);
     if(fx[x]<fx[root])    root=x;
 }
 void getsz(int x,int fa)
 //处理出当前连通块的大小
 {
     sz[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getsz(e[k].to,x);
             sz[x]+=sz[e[k].to];
         }
 }
 void getdeep(int u,int fa)
 {
     tmp[++tmp[]]=dep[u];
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             dep[e[k].to]=dep[u]+e[k].d;
             getdeep(e[k].to,u);
         }
 }
 int cal(int u,int cost)
 {
     dep[u]=cost;tmp[]=;
     getdeep(u,);
     sort(tmp+,tmp+tmp[]+);
     int l,r,sum=;
     r=tmp[]+;
     for(l=;l<=tmp[];l++)
     {
         while(r>&&tmp[r-]>k-tmp[l])    r--;
         if(l>=r)    break;
         sum+=(r-l-);
     }
     return sum;
 }
 void solve(int u)
 //解决u所在连通块的问题，u作为该块的根
 {
     ans+=cal(u,);vis[u]=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             ans-=cal(e[k].to,e[k].d);
             root=;
             getsz(e[k].to,);sum=sz[e[k].to];
             getroot(e[k].to,);
             solve(root);
         }
 }
 int main()
 {
     int i,u,v,l;
     fx[]=0x3f3f3f3f;
     while()
     {
         memset(f1,,sizeof(f1));ne=ans=;
         memset(vis,,sizeof(vis));
         scanf("%d%d",&n,&k);
         if(n==&&k==)    break;
         for(i=;i<n;i++)
         {
             scanf("%d%d%d",&u,&v,&l);
             e[++ne].to=v;e[ne].nxt=f1[u];e[ne].d=l;f1[u]=ne;
             e[++ne].to=u;e[ne].nxt=f1[v];e[ne].d=l;f1[v]=ne;
         }
         sum=n;
         getroot(,);
         solve(root);
         printf("%d\n",ans);
     }
     return ;
 }