bzoj1468 Tree

最新推荐文章于 2018-07-29 21:12:00 发布

anchoupi8353

最新推荐文章于 2018-07-29 21:12:00 发布

阅读量88

点赞数

原文链接：http://www.cnblogs.com/wjyi/p/5615991.html

版权

本文深入解析了一道经典的点分治题目，通过详细的代码解释了如何在递归子树时减去在算父亲时的不合法方案，提供了一个解决树上点对距离问题的有效方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

bzoj1468 Tree

最经典的点分治题目，在递归子树的时候减去在算父亲时的不合法方案。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<queue>
 5 #include<algorithm>
 6 #include<cmath>
 7 #define ll long long
 8 #define N 40005
 9 using namespace std;
10 struct Node{
11   int to,next,v;
12 }e[N<<1];
13 int sum,rt,n,k,head[N],tot,cnt,mx[N],sz[N],tmp[N],ans;
14 bool vis[N];
15 void add(int x,int y,int z){
16   e[++tot]=(Node){y,head[x],z};head[x]=tot;
17   e[++tot]=(Node){x,head[y],z};head[y]=tot; 
18 }
19 void getroot(int x,int fa){
20   mx[x]=0;sz[x]=1;
21   for(int i=head[x];i;i=e[i].next)if(e[i].to!=fa&&!vis[e[i].to]){
22     getroot(e[i].to,x);
23     sz[x]+=sz[e[i].to];
24     mx[x]=max(mx[x],sz[e[i].to]);
25   }mx[x]=max(mx[x],sum-sz[x]);
26   if(mx[x]<mx[rt])rt=x;
27 }
28 void getdis(int x,int fa,int d){
29   tmp[++cnt]=d;
30   for(int i=head[x];i;i=e[i].next)if(e[i].to!=fa&&!vis[e[i].to]){
31     getdis(e[i].to,x,d+e[i].v);
32   }
33 }
34 void calc(int x,int dis,int type){
35   cnt=0;
36   getdis(x,0,0);
37   sort(tmp+1,tmp+1+cnt);
38   int l=1,r=cnt;
39   while(l<=r){
40     while(tmp[l]+tmp[r]>dis)r--;
41     if(l>r)break;
42     ans+=(r-l)*type;l++;
43   }
44 }
45 void work(int x){
46   vis[x]=1;
47   calc(x,k,1);
48   for(int i=head[x];i;i=e[i].next)if(!vis[e[i].to]){
49     calc(e[i].to,k-e[i].v*2,-1);
50     sum=sz[e[i].to];rt=0;
51     getroot(e[i].to,0);
52     work(rt);
53   }
54 }
55 int main(){
56 //  freopen("test.in","r",stdin);
57   scanf("%d",&n); 
58   for(int i=1;i<n;i++){
59     int x,y,z;scanf("%d%d%d",&x,&y,&z);
60     add(x,y,z);
61   }
62   scanf("%d",&k);
63   sum=mx[0]=n;rt=0;ans=0;
64   getroot(1,0);
65   work(rt);
66   printf("%d\n",ans);
67 }