洛谷 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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值