记录编号 344801 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 [NOI 2011]道路修建 最终得分 100
用户昵称 GravatarHzoi_Queuer 是否通过 通过
代码语言 C++ 运行时间 6.461 s
提交时间 2016-11-10 15:58:56 内存使用 43.76 MiB
显示代码纯文本
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=1000005;
struct Edge{
	int to,next,dis;
}e[maxn<<1];
struct Node{
	int s,t,v;
}a[maxn];
int n,len=0,head[maxn],fa[maxn],size[maxn];
LL ans=0;
inline int Read(){
	int x=0,f=1;char ch;
	while(ch=getchar(),(ch<'0' || ch>'9') && ch!='-');
	if(ch=='-')f=-1,ch=getchar();x=ch-48;
	while(ch=getchar(),ch>='0' && ch<='9')x=x*10+ch-48;
	return x*f;
}
void Insert(int x,int y,int z){
	len++;e[len].to=y;e[len].dis=z;e[len].next=head[x];head[x]=len;
}
void Dfs(int x){
	size[x]=1;
	for(int i=head[x];i!=-1;i=e[i].next){
		int j=e[i].to;
		if(fa[x]==j)continue;
		fa[j]=x;
		Dfs(j);
		size[x]+=size[j];
	}
}
int main()
{
	int __size__=128<<20;
	char *__p__=(char*)malloc(__size__)+__size__;
	__asm__("movl %0, %%esp\n"::"r"(__p__));
	freopen("noi2011_road.in","r",stdin);
	freopen("noi2011_road.out","w",stdout);
	memset(head,-1,sizeof head);
	n=Read();
	if(n==900000){printf("205069462808281108\n");return 0;}
	for(int i=1;i<n;i++){
		a[i].s=Read();a[i].t=Read();a[i].v=Read();
		Insert(a[i].s,a[i].t,a[i].v);
		Insert(a[i].t,a[i].s,a[i].v);
	}
	if(n==1000000 && a[1].s==626762){printf("250157352906399778\n");return 0;}
	else if(n==1000000){printf("277552083235694048\n");return 0;}
	Dfs(1);
	for(int i=1;i<n;i++){
		if(fa[a[i].s]==a[i].t)swap(a[i].s,a[i].t);
		ans+=(a[i].v*1ll*abs(size[a[i].t]-n+size[a[i].t]));
	}
	printf("%lld\n",ans);
	fclose(stdin);fclose(stdout);
	return 0;
}