码迷,mamicode.com
首页 > 其他好文 > 详细

2435: [Noi2011]道路修建(树上操作)

时间:2018-03-10 12:01:53      阅读:165      评论:0      收藏:0      [点我收藏+]

标签:lin   math   cst   target   font   题解   nbsp   大小   答案   

2435: [Noi2011]道路修建

题目:传送门


 

题解:

   建完边之后以1为根建树,统计深度和各个点的子树大小(包括自己)

   询问的时候:答案=长度*abs(n-深度大的点的子树大小*2)

   ans+=a[i].c*abs(n-tot[y]*2)

 


代码:

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<cstdlib>
 4 #include<cmath>
 5 #include<algorithm>
 6 using namespace std;
 7 typedef long long LL;
 8 struct node
 9 {
10     int x,y,c,next;
11 }a[2110000];int len,last[1110000];
12 int n,tot[1110000],dep[1110000];
13 void ins(int x,int y,int c)
14 {
15     len++;a[len].x=x;a[len].y=y;a[len].c=c;
16     a[len].next=last[x];last[x]=len;
17 }
18 void dfs(int x,int fa)
19 {
20     tot[x]=1;dep[x]=dep[fa]+1;
21     for(int k=last[x];k;k=a[k].next)
22     {
23         int y=a[k].y;
24         if(y!=fa)
25         {
26             dfs(y,x);
27             tot[x]+=tot[y];
28         }
29     }
30 }
31 int main()
32 {
33     scanf("%d",&n);len=0;memset(last,0,sizeof(last));
34     for(int i=1;i<n;i++)
35     {
36         int x,y,c;scanf("%d%d%d",&x,&y,&c);
37         ins(x,y,c);ins(y,x,c);
38     }
39     dfs(1,0);
40     LL ans=0;
41     for(int i=1;i<=len;i+=2)
42     {
43         int x=a[i].x,y=a[i].y;
44         if(dep[x]>dep[y])swap(x,y);
45         ans+=LL(a[i].c)*abs(n-tot[y]*2);
46     }
47     printf("%lld\n",ans);
48     return 0;
49 }

 

2435: [Noi2011]道路修建(树上操作)

标签:lin   math   cst   target   font   题解   nbsp   大小   答案   

原文地址:https://www.cnblogs.com/CHerish_OI/p/8537340.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!