bzoj1468

1468: Tree

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 1023  Solved: 532
[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

 

Source

题解：

　　点分治裸题，那么什么是点分治呢

　　点分治

　　1、求树的重心
　　2、计算以当前重心为根的子树的答案
　　3、去掉以当前重心儿子为根的子树的答案
　　4、枚举每个儿子，分治

代码：

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<algorithm>
 #include<cmath>
 #define maxn 40005
 using namespace std;
 int n,k,i,j,tot,root,ans,L,all;
 int pre[maxn*],now[maxn],v[maxn*],val[maxn*],son[maxn],f[maxn];
 int dist[maxn],data[maxn];
 bool vis[maxn];
 int read()
 {
     int x=; char ch; bool bo=;
     while (ch=getchar(),ch<''||ch>'') if (ch=='-') bo=;
     while (x=x*+ch-'',ch=getchar(),ch>=''&&ch<='');
     if (bo) return -x; return x;
 }
 void ins(int a,int b,int c){
     pre[++tot]=now[a]; now[a]=tot; v[tot]=b; val[tot]=c;
 }
 void get_root(int u,int fa)
 {
     son[u]=; f[u]=;
     for (int p=now[u]; p; p=pre[p])
     {
         int vv=v[p];
         if (vv==fa || vis[vv]) continue;
         get_root(vv,u); son[u]+=son[vv];
         f[u]=max(f[u],son[vv]);
     }
     f[u]=max(f[u],all-son[u]);
     if (f[u]<f[root]) root=u;
 }
 void get_array(int u,int fa)
 {
     data[++L]=dist[u];
     for (int p=now[u]; p; p=pre[p])
     {
         int vv=v[p];
         if (vv!=fa&&!vis[vv])
         dist[vv]=dist[u]+val[p],get_array(vv,u);
     }
 }
 int calc(int u,int value)
 {
     dist[u]=value; L=; get_array(u,-);
     sort(data+,data++L);
     int preans=,l=,r=L;
     while (l<r)
     {
         if (data[l]+data[r]<=k) preans+=(r-l),l++; else r--;
     }
     return preans;
 }
 void solve(int u)
 {
     ans+=calc(u,); vis[u]=;
     //cout<<"        ans       "<<u<<endl;
     for (int p=now[u]; p; p=pre[p])
     {
         int vv=v[p];
         //cout<<"                   "<<vv<<endl;
         if (vis[vv]) continue;
         ans-=calc(vv,val[p]); f[root=]=n+;
         all=son[vv]; get_root(vv,-);
         solve(root);
     }
 }
 int main()
 {
     n=read();
     for (int i=; i<=n-; i++)
     {
         int u=read(),v=read(),value=read();
         ins(u,v,value); ins(v,u,value);
     }
     k=read();
     f[root=]=n+; all=n;
     get_root(,-);
     //cout<<"                "<<root<<endl;
     solve(root);
     printf("%d\n",ans);
     return ;
 }