树形问题--奶牛大集会

P3689奶牛大集会

时间限制 : - MS   空间限制 : 65536 KB  

评测说明 : 时限1000ms

问题描述

Bessie正在计划一年一度的奶牛大集会，来自全国各地的奶牛将来参加这一次集会。当然，她会选择最方便的地点来举办这次集会。每个奶牛居住在 N(1<=N<=100,000) 个农场中的一个，这些农场由N-1条道路连接，并且从任意一个农场都能够到达另外一个农场。道路i连接农场A_i和B_i(1 <= A_i <=N; 1 <= B_i <= N),长度为L_i(1 <= L_i <= 1,000)。集会可以在N个农场中的任意一个举行。另外，每个牛棚中居住者C_i(0 <= C_i <= 1,000)只奶牛。在选择集会的地点的时候，Bessie希望最大化方便的程度(也就是最小化不方便程度)。比如选择第X个农场作为集会地点，它的不方便程度是其它牛棚中每只奶牛去参加集会所走的路程之和，(比如，农场i到达农场X的距离是20，那么总路程就是C_i*20)。帮助Bessie找出最方便的地点来举行大集会。 考虑一个由五个农场组成的国家，分别由长度各异的道路连接起来。在所有农场中，3号和4号没有奶牛居住。

输入格式

第一行：一个整数N 
第二到N+1行：第i+1行有一个整数C_i 
第N+2行到2*N行，第i+N+1行为3个整数：A_i,B_i和L_i。

输出格式

第一行：一个值，表示最小的不方便值


样例输入


5
1
1
0
0
2
1 3 1
2 3 2
3 4 3
4 5 3


样例输出


15


来源  Usaco2010 Mar

分析：

首先根据题意建好一棵树（根为哪个节点无所谓，建议采用存边的方式）.开两个数组f[n],f2[n]

f[i].cost表示把以i为根的子树上的所有牛集中到i号点的花费

f[i].num表示把以i为根的子树上的牛的总数

f2[i]表示把除了以i为根的子树上的所有牛集中到i号点的花费

此处的f我处理为结构体,方便记下费用和牛的个数两个值（f[i].cost和f[i].num顾名思义）

我们可以在求f[i]的时候顺便把以i为根的子树上的牛的个数求了

f[i].cost = sum ( f[j].cost + f[j].num * len[i][j] )  

f[i].num = sum ( f[j].num) + c[i] 

(j是i的儿子)

那么f很快就求完了，难点在于f2

假设i为j的父亲，

f2[j]=f2[i]+f[i]-f[j]-f[j].num*len[i][j]+(sum-f[j].num)*len[i][j]（sum为牛的总数）            （想一想，为什么？）

(j是i的儿子)

那么答案是什么呢？

ans = min ( f[i].cost+f2[i] )

AC <code>

(rude words used!!!)

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define inf 10000000000000LL
using namespace std;
struct line{
	long long from,to,len;
};
line edge[1000005];
struct node{
	long long cost;
	long long num;
};
long long last[100005],__next[1000005];
long long n,m=0,sum=0;
long long c[100005];
node f[100005];
node f2[100005];
void add_edge(long long x,long long y,long long len){
	m++;
	__next[m]=last[x];
	last[x]=m;
	edge[m].from=x;
	edge[m].to=y;
	edge[m].len=len;
}
void fuck_down(long long x,long long fa){
	long long i,j,k;
	for(j=last[x];j;j=__next[j]){
		i=edge[j].to;
		if(i==fa)continue;
		fuck_down(i,x);
		f[x].num+=f[i].num;
		f[x].cost+=f[i].cost+f[i].num*edge[j].len;
	}
}
void fuck_up(long long i,long long fa){
	long long h,j,k;
	for(h=last[i];h;h=__next[h]){
		j=edge[h].to;
		if(j==fa)continue;
		f2[j].cost=f2[i].cost+f[i].cost-f[j].cost-f[j].num*edge[h].len+(sum-f[j].num)*edge[h].len;
		fuck_up(j,i);
	}
}
int main(){
	long long i,j,k,ans=inf;
	cin>>n;
	for(i=1;i<=n;i++){
		scanf("%I64d",&c[i]);
		f[i].num=c[i];
		sum+=c[i];
	}
	for(i=1;i<n;i++){
		long long x,y,l;
		scanf("%I64d%I64d%I64d",&x,&y,&l);
		add_edge(x,y,l);
		add_edge(y,x,l);
	}
	fuck_down(1,0);
	fuck_up(1,0);
	for(i=1;i<=n;i++){
		ans=min(ans,f[i].cost+f2[i].cost);
	}
	cout<<ans;
}