BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会 树形DP + 带权重心
生活随笔
收集整理的這篇文章主要介紹了
BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会 树形DP + 带权重心
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
Description
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號沒有奶牛居住。Input
第一行:一個整數N * 第二到N+1行:第i+1行有一個整數C_i * 第N+2行到2*N行,第i+N+1行為3個整數:A_i,B_i和L_i。Output
* 第一行:一個值,表示最小的不方便值。 題解:不難發現選定點必須是樹的帶權重心 (任意兒子大小不能超過總和一半).? 直接 DFS 求重心并算一下貢獻即可.? #include <bits/stdc++.h> #define setIO(s) freopen(s".in","r",stdin) #define maxn 1000000 #define inf 100000000000000 #define ll long long using namespace std; ll f[maxn], siz[maxn], sumv[maxn]; int C[maxn], hd[maxn], to[maxn << 1], nex[maxn << 1], val[maxn << 1]; int n, edges, root; ll tot; void add(int u, int v, int c) {nex[++edges] = hd[u], hd[u] = edges, to[edges] = v, val[edges] = c; } void dfs(int u, int ff) {siz[u] = C[u], f[u] = 0; for(int i = hd[u]; i ; i = nex[i]) {int v = to[i]; if(v == ff) continue; dfs(v, u); siz[u] += siz[v];f[u] = max(f[u], siz[v]); } f[u] = max(f[u], tot - siz[u]); if(f[u] < f[root]) root = u; } void calc(int u,int ff) {siz[u] = C[u], sumv[u] = 0; for(int i = hd[u]; i ; i = nex[i]){int v = to[i]; if(v == ff) continue; calc(v, u); siz[u] += siz[v]; sumv[u] += sumv[v] + val[i] * siz[v]; } } int main() {// setIO("input"); scanf("%d",&n); for(int i = 1; i <= n; ++i) scanf("%d",&C[i]), tot += C[i]; for(int i = 1, u, v, c; i < n; ++i) {scanf("%d%d%d",&u,&v,&c); add(u, v, c); add(v, u, c); }f[0] = inf, root = 0, dfs(1, 0);memset(siz, 0, sizeof(siz)), calc(root, 0); printf("%lld\n",sumv[root]); return 0; }
轉載于:https://www.cnblogs.com/guangheli/p/10982700.html
總結
以上是生活随笔為你收集整理的BZOJ 1827: [Usaco2010 Mar]gather 奶牛大集会 树形DP + 带权重心的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 海康相机IP搜索协议研究
- 下一篇: 双缓冲 android,Android