洛谷 P1364 医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为1。如上图中，

若医院建在1 处，则距离和=4+12+2*20+2*40=136；若医院建在3 处，则距离和=4*2+13+20+40=81……

输入输出格式

输入格式：

 

第一行一个整数n，表示树的结点数。(n≤100)

接下来的n行每行描述了一个结点的状况，包含三个整数，整数之间用空格(一个或多个)分隔，其中：第一个数为居民人口数；第二个数为左链接，为0表示无链接；第三个数为右链接。

 

输出格式：

 

一个整数，表示最小距离和。

 

输入输出样例

输入样例#1： 复制

5						
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0

输出样例#1： 复制

81

（这个范围怎么写都过得去，然后枚举每个点求一下，dfs处理距离）

vector<int>v[maxn];
void dfs(int id,int x,int fa,int step){
    for(auto d:v[x]){
        if(d == fa) continue;ans[id] += step * a[d];
        dfs(id,d,x,step + 1);
    }
}
int main() { int m,t;
    ios::sync_with_stdio(false);
    while(cin >> n){
        for(int i = 1;i <= n;i++){
            int l,r;
            cin >> a[i] >> l >> r;
            if(l != 0) v[l].push_back(i),v[i].push_back(l);
            if(r != 0) v[i].push_back(r),v[r].push_back(i);
        }
        for(int i = 1;i <= n;i++){
//                ans[i] = a[i];
                dfs(i,i,0,1);
        }
        ll Min = 1e18;
        for(int i = 1;i <= n;i++) Min = min(Min,ans[i]);
        cout << Min << endl;
    }
    cerr << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}

下面我们看一下On怎么来解决这个题，其实虽然在最短路专题做的，上来一看肯定是树形dp呀，

sum_i代表为i为根节点的子树权值和，f_i代表以i为根节点所有子树的ans和，

u是v的父节点，考虑转移，显然可以得到

f[v] = f[u] + (sum[1] - sum[u]) - sum[u] 

然后遍历一下即可

#include <bits/stdc++.h>
#include <time.h>
#define fi first
#define se second
#define endll "\n"
#define MS0(X) memset((X), 0, sizeof((X)))
#define MS1(X) memset((X), -1, sizeof((X)))
#define LEN(X) strlen(X)
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
///vector(len,val);
using namespace std;

typedef long long ll;
typedef double db;
int xx[4] = {1,-1,0,0};
int yy[4] = {0,0,1,-1};
const double eps = 1e-9;
typedef pair<int,int>  P;
const int maxn = 2e6 + 5000;
const ll mod = 1e9 + 7;
inline int sign(db a) { return a < -eps ? -1 : a > eps;}
inline int cmp(db a,db b){ return sign(a - b);}
void debug(int x){  cout << x << endl; }
ll mul(ll a,ll b,ll c) { ll res = 1; while(b) {  if(b & 1) res *= a,res %= c;  a *= a,a %= c,b >>= 1;  }  return res;}
ll phi(ll x) {  ll res = x;  for(ll i = 2; i * i <= x; i++) { if(x % i == 0) res = res / i * (i - 1);   while(x % i == 0) x /= i;   }  if(x > 1) res = res / x  * (x - 1);    return res;}

int fa[maxn];
int Find(int x) {   if(x != fa[x]) return fa[x] = Find(fa[x]);  return fa[x];}
ll c,n,k;
ll ans[maxn];
ll a[maxn];
vector<int>v[maxn];
ll sum[maxn],f[maxn];///f_i 代表以i为根，到所有子树的权值和
void dfs(int x,int fa,int step){
    for(auto d:v[x]){
        if(d == fa) continue; f[1] += a[d] * step,dfs(d,x,step + 1); sum[x] += sum[d] + a[d];

    }
}
ll s = 0;
void dfss(int x,int fa){
    for(auto d:v[x]){
        if(d == fa) continue;
        f[d] = f[x] - sum[d] + (s - sum[d]);
        dfss(d,x);
    }
}
int main() { int m,t;
    ios::sync_with_stdio(false);
    while(cin >> n){
        for(int i = 1;i <= n;i++){
            int l,r;
            cin >> a[i] >> l >> r; s += a[i];
            if(l != 0) v[l].push_back(i),v[i].push_back(l);
            if(r != 0) v[i].push_back(r),v[r].push_back(i);
        }
        dfs(1,0,1);
        for(int i = 1;i <= n;i++) sum[i] += a[i];
        dfss(1,0);
        ll Min = 1e18;
        for(int i = 1;i <= n;i++) Min = min(Min,f[i]);
        cout << Min << endl;
    }
    cerr << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}