【2018/07/16测试T2】【SDOJ 2509】奶牛大集会

【题目】

题目描述：

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号没有奶牛居住。

      1     3     4     5
      @--1--@--3--@--3--@[2]
     [1]    |
            2
            |
            @[1]
            2

Bessie可以在五个农场中的任意一个举办集会，下面就是在每个位置举办集会的不方便值的 统计表。

  集会地点    -----  不方便程度  ------
              B1  B2  B3  B4  B5   Total
     1         0   3   0   0  14    17
     2         3   0   0   0  16    19
     3         1   2   0   0  12    15
     4         4   5   0   0   6    15
     5         7   8   0   0   0    15

如果Bessie在农场1举办集会，那么每个农场各自的不方便值分别是

      农场 1     0 -- 到达不需要时间!
      农场 2     3 -- 总的距离是 2+1=3  x 1 奶牛 = 3
      农场 3     0 -- 没奶牛!
      农场 4     0 -- 没奶牛!
      农场 5    14 -- 总的距离是 3+3+1=7 x 2 奶牛 = 14

因此，总的不方便值是17。

最小的不方便值是15，当在3号，4号或者5号农场举办集会的时候。

输入格式：

第一行：一个整数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

备注：

数据规模与约定：

30%数据：N<=30

100%数据：N<=100000，0<=Ci，Wi<=1000

 

【分析】

本题要用到的知识点：树形DP+容斥原理+换根

定义f[i]表示以i为关键点的答案

dis[i]表示以i到1的距离

num[i]表示以i为根的子数点权之和，tot为总点权和

假设现在dp到i号节点，j为i的一个儿子，w为此时的边权

那么f[j]=f[i]+（tot-num[j]）*w-num[j]*w；

由父节点转移到子节点，容斥会比较难想一点，建议画图理解

 

【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100005,M=200005;
int n,t;
int c[N],first[N];
int v[M],w[M],next[M];
long long ans,tot,f[N],num[N],dis[N];
void add(int x,int y,int z)
{
	t++;
	next[t]=first[x];
	first[x]=t;
	v[t]=y;
	w[t]=z;
}
void dfs(int x,int father)
{
	int i,j;
	num[x]+=c[x];
	for(i=first[x];i;i=next[i])
	{
		j=v[i];
		if(j==father)
		  continue;
		dis[j]=dis[x]+w[i];
		dfs(j,x);
		num[x]+=num[j];
	}
}
void dp(int x,int father)
{
	int i,j;
	for(i=first[x];i;i=next[i])
	{
		j=v[i];
		if(j==father)
		  continue;
		f[j]=f[x]+(tot-num[j])*w[i]-num[j]*w[i];
		ans=min(ans,f[j]);
		dp(j,x);
	}
}
int main()
{
	int x,y,z,i;
	scanf("%d",&n);
	for(i=1;i<=n;++i)
	{
		scanf("%d",&c[i]);
		tot+=c[i];
	}
	for(i=1;i<n;++i)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	dis[1]=0;
	dfs(1,0);
	for(i=1;i<=n;++i)
	  f[1]+=1ll*dis[i]*c[i];
	ans=f[1];
	dp(1,0);
	printf("%lld",ans);
	return 0;
}