【BZOJ】【P2435】【Noi2011】【道路修建】【题解】【dfs】

最新推荐文章于 2017-09-04 16:42:00 发布

iamzky

最新推荐文章于 2017-09-04 16:42:00 发布

阅读量1.1w

点赞数

CC 4.0 BY-SA版权

分类专栏： OI 文章标签： bzoj

本文链接：https://blog.csdn.net/iamzky/article/details/43198477

OI 专栏收录该内容

324 篇文章

订阅专栏

本文介绍了一种使用DFS算法解决特定问题的方法，并详细解释了如何通过代码实现避免栈溢出的问题。其中包括输入解析、图构建、深度优先搜索过程以及结果计算。

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=2435

说好的卡爆栈呢

Code;

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
struct edge{int v,w;};
vector<edge>G[maxn];
typedef long long LL;
LL sum=0,ans=0;
int size[maxn],vis[maxn],n;
void dfs(int u){
	size[u]=1;edge e;
	for(int i=0;i<G[u].size();i++)
	if(!vis[(e=G[u][i]).v]){
		vis[e.v]=1;
		dfs(e.v);
		size[u]+=size[e.v];
		ans+=(LL)abs(n-size[e.v]-size[e.v])*e.w;
	}
}
int getint(){
	int res=0;char c=getchar();
	while(!isdigit(c))c=getchar();
	while(isdigit(c))res=res*10+c-'0',c=getchar();
	return res;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u=getint(),v=getint(),w=getint();
		G[u].push_back((edge){v,w});
		G[v].push_back((edge){u,w});
	}vis[1]=1;dfs(1);
	cout<<ans<<endl;
	return 0;
}