首页 > 代码库 > 【DFS】bzoj2435 [Noi2011]道路修建
【DFS】bzoj2435 [Noi2011]道路修建
两遍DFS。第一遍统计以每个点为根的子树大小,第二遍更新答案。
1 #include<cstdio> 2 #include<iostream> 3 using namespace std; 4 int v[2000001],w[2000001],first[2000001],next[2000001],en,sz[1000001]; 5 bool vis[1000001]; 6 long long ans; 7 inline int Abs(const int &x){return x<0 ? -x : x;} 8 inline void AddEdge(const int &U,const int &V,const int &W) 9 {v[++en]=V;w[en]=W;next[en]=first[U];first[U]=en;}10 int n,a,b,c,res;11 char C;12 inline int Get()13 {14 res=0;C=‘*‘;15 while(C<‘0‘||C>‘9‘)C=getchar();16 while(C>=‘0‘&&C<=‘9‘){res=res*10+(C-‘0‘);C=getchar();}17 return res;18 }19 void dfs(int cur)20 {21 vis[cur]=true;22 sz[cur]=1;23 for(int i=first[cur];i;i=next[i])24 if(!vis[v[i]]){dfs(v[i]);sz[cur]+=sz[v[i]];}25 vis[cur]=false;26 }27 void dfs2(int cur)28 {29 vis[cur]=true;30 for(int i=first[cur];i;i=next[i])31 if(!vis[v[i]])32 {33 ans+=(long long)Abs((sz[v[i]]<<1)-n)*w[i];34 dfs2(v[i]);35 }36 vis[cur]=false;37 }38 int main()39 {40 n=Get();41 for(int i=1;i<n;i++){a=Get();b=Get();c=Get();AddEdge(a,b,c);AddEdge(b,a,c);}42 dfs(1);43 dfs2(1);44 cout<<ans<<endl;45 return 0;46 }
【DFS】bzoj2435 [Noi2011]道路修建
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。