【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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值