C++之路进阶——bzoj1468（tree）

F.A.Qs	Home	Discuss	ProblemSet	Status	Ranklist	Contest	ModifyUser  gryz2016	Logout	捐赠本站

Notice:由于本OJ建立在Linux平台下，而许多题的数据在Windows下制作，请注意输入、输出语句及数据类型及范围，避免无谓的RE出现。

1468: Tree

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 774  Solved: 412
[Submit][Status][Discuss]

Description

给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K

Input

N（n<=40000） 接下来n-1行边描述管道，按照题目中写的输入 接下来是k

Output

一行，有多少对点之间的距离小于等于k

Sample Input

7
1 6 13
6 3 9
3 5 7
4 1 3
2 4 20
4 7 2
10

Sample Output

5

HINT

题解：

点分。。。

代码：

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #define maxn  400100

 using namespace std;

 int n,m,root,sum,ans,cnt,head[maxn],son[maxn],f[maxn],vis[maxn],deep[maxn],d[maxn];

 struct ss
    {
         int to;
         int next;
         int w;
    }e[maxn];

 void insert(int u,int v,int w)
    {
        e[++cnt].to=v; e[cnt].next=head[u]; e[cnt].w=w; head[u]=cnt;
        e[++cnt].to=u; e[cnt].next=head[v]; e[cnt].w=w; head[v]=cnt;
    }

 void getroot(int x,int fa)
    {
         son[x]=;
         f[x]=;
         for (int i=head[x];i;i=e[i].next)
           if (!vis[e[i].to]&&e[i].to!=fa)
              {
                   getroot(e[i].to,x);
                   son[x]+=son[e[i].to];
                   f[x]=max(f[x],son[e[i].to]);
              }
       f[x]=max(f[x],sum-son[x]);
       if (f[x]<f[root]) root=x;
      } 

 void getdeep(int x,int fa)
      {
          deep[++deep[]]=d[x];
          for (int i=head[x];i;i=e[i].next)
              if (!vis[e[i].to]&&e[i].to!=fa)
                  {
                    d[e[i].to]=d[x]+e[i].w;
                    getdeep(e[i].to,x);
                  }
            }

 int cal(int x,int now)
     {
       d[x]=now;
       deep[]=;
       getdeep(x,);
       sort(deep+,deep+deep[]+);
       int t=,l,r;
       for(l=,r=deep[];l<r;)
        {
         if(deep[l]+deep[r]<=m){t+=r-l;l++;}
         else r--;
        }
     return t;
     }

 void work(int x)
     {
      ans+=cal(x,);
      vis[x]=;
      for (int i=head[x];i;i=e[i].next)
        if(!vis[e[i].to])
          {
            ans-=cal(e[i].to,e[i].w);
            sum=son[e[i].to];
            root=;
            getroot(e[i].to,root);
            work(root);
           }
     }

 int main()
    {
       scanf("%d",&n);
          for (int i=;i<n;i++)
            {
               int u,v,w;
               scanf("%d%d%d",&u,&v,&w);
               insert(u,v,w);
            }
       scanf("%d",&m);
        sum=n;
        f[]=0x7fffffff;
        getroot(,);
        work();
        printf("%d",ans);
    }