[Usaco2010 Mar]gather 奶牛大集會
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号沒有奶牛居住。
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLicGcq5yNygTMvw1cldWYtl2LcVmbpxmbPV2ZkVnSvwVbvNmL5NHZ5xmL3d3dvw1LcpDc0RHaiojIsJye.jpg)
分析:
先以1(随便)為根dfs一次,求出以每個節點為根時,他所在的子樹的人數個數sz,并且計算出以1為根時的不友善度。
第二次時,繼續以1為根,這時假設目前節點為x,不友善度為cost,兒子節點為y。把x的不友善度cost向y轉化,其實就是
cost+(sz[1]-2*sz[y])*edge[i].cost (畫個圖就知道了。。。)
#include <set>
#include <map>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <string>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define debug puts("here")
#define rep(i,n) for(int i=0;i<n;i++)
#define rep1(i,n) for(int i=1;i<=n;i++)
#define REP(i,a,b) for(int i=a;i<=b;i++)
#define foreach(i,vec) for(unsigned i=0;i<vec.size();i++)
#define pb push_back
#define RD(n) scanf("%d",&n)
#define RD2(x,y) scanf("%d%d",&x,&y)
#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define RD4(x,y,z,w) scanf("%d%d%d%d",&x,&y,&z,&w)
#define All(vec) vec.begin(),vec.end()
#define MP make_pair
#define PII pair<int,int>
#define PQ priority_queue
#define cmax(x,y) x = max(x,y)
#define cmin(x,y) x = min(x,y)
#define Clear(x) memset(x,0,sizeof(x))
/*
#pragma comment(linker, "/STACK:1024000000,1024000000")
int size = 256 << 20; // 256MB
char *p = (char*)malloc(size) + size;
__asm__("movl %0, %%esp\n" :: "r"(p) );
*/
/******** program ********************/
const int MAXN = 1e5+5;
const ll INF = 1e15;
struct Edge{
int y,cost,next;
}edge[MAXN<<1];
int c[MAXN],n;
ll dp[MAXN],sz[MAXN],ans[MAXN];
int po[MAXN],tol;
inline void add(int x,int y,int cost){
edge[++tol].y = y;
edge[tol].cost = cost;
edge[tol].next = po[x];
po[x] = tol;
}
void dfsDp(int x,int fa){
dp[x] = 0;
sz[x] = c[x];
for(int i=po[x];i;i=edge[i].next){
int y = edge[i].y;
if(y==fa)continue;
dfsDp(y,x);
sz[x] += sz[y];
dp[x] += sz[y]*edge[i].cost+dp[y];
}
}
void dfsAns(int x,int fa,ll cost){
ans[x] = cost;
for(int i=po[x];i;i=edge[i].next){
int y = edge[i].y;
if(y==fa)continue;
ll tmp = cost+(sz[1]-2*sz[y])*edge[i].cost;
dfsAns(y,x,tmp);
}
}
int main(){
#ifndef ONLINE_JUDGE
freopen("sum.in","r",stdin);
//freopen("sum.out","w",stdout);
#endif
while(~RD(n)){
rep1(i,n)
RD(c[i]);
int x,y,cost;
REP(i,2,n){
RD3(x,y,cost);
add(x,y,cost);
add(y,x,cost);
}
dfsDp(1,0);
dfsAns(1,0,dp[1]);
ll tmp = INF;
rep1(i,n)
cmin(tmp,ans[i]);
cout<<tmp<<endl;
}
return 0;
}
轉載于:https://www.cnblogs.com/yejinru/p/3295971.html