poj - 4045 - Power Station

题意：一棵有n个结点的树，要取其中的一个结点，使得该结点到其他所有结点的距离和dis最小，即损耗I ＊ I ＊ R ＊ dis最小，输出最小损耗和该结点（有多个的话按结点编号从小到大输出）（3 <= n <= 50000, 1 <= I <= 10, 1 <= R <= 50）。

题目链接：http://poj.org/problem?id=4045

——>>怒刷树状dp。。。

设cnt[i]为以i为根的子树的结点数，d[i]为以i为根的子树中所有结点到i的距离和，一次dfs求出*cnt和*d，则

状态转移方程为：cnt[x] += cnt[v[e]];

d[x] += d[v[e]] + cnt[v[e]];（v[e]为x的子结点）

设f[i]为以i为根，刚才dfs中i的父结点为孩子，新生子树中所有结点到i的距离和，再一次dfs，即dp，则

状态转移方程为：f[x] = d[fa] - d[x] - cnt[x] + f[fa] + n －cnt[x] = d[fa] - d[x] - 2 * cnt[x] + f[fa] + n;（x为第一次dfs中fa的子结点）。

注意：用64位整数。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 50000 + 10;
const int INF = 0x3f3f3f3f;
int n, I, R, head[maxn], nxt[maxn<<1], u[maxn<<1], v[maxn<<1], ecnt, cnt[maxn];
long long d[maxn], f[maxn];

void init(){
ecnt = 0;
memset(head, -1, sizeof(head));
}

void addEdge(int uu, int vv){
u[ecnt] = uu;
v[ecnt] = vv;
nxt[ecnt] = head[uu];
head[uu] = ecnt;
ecnt++;
}

void read(){
int i, uu, vv;
scanf("%d%d%d", &n, &I, &R);
for(i = 0; i < n-1; i++){
scanf("%d%d", &uu, &vv);
addEdge(uu, vv);
addEdge(vv, uu);
}
}

void dfs(int x, int fa){
d[x] = 0;
for(int e = head[x]; e != -1; e = nxt[e]) if(v[e] != fa){
dfs(v[e], x);
cnt[x] += cnt[v[e]];
d[x] += d[v[e]] + cnt[v[e]];
}
}

void dp(int x, int fa){
if(x == 1) f[x] = 0;
else f[x] = d[fa] - d[x] - 2 * cnt[x] + f[fa] + n;
for(int e = head[x]; e != -1; e = nxt[e]) if(v[e] != fa) dp(v[e], x);
}

void solve(){
int i, flag = 1;
long long Min = INF;
for(int i = 1; i <= n; i++) cnt[i] = 1;
dfs(1, -1);
dp(1, -1);
for(i = 1; i <= n; i++) Min = min(Min, d[i] + f[i]);
printf("%I64d\n", I * I * R * Min);
for(i = 1; i <= n; i++) if(d[i] + f[i] == Min) {
if(flag) flag = 0;
else putchar(' ');
printf("%d", i);
}
puts("\n");
}

int main()
{
int T;
scanf("%d", &T);
while(T--){
init();
read();
solve();
}
return 0;
}