$/text{Sol. Luogu P1642}$ 规划


/(/text{Sol. Luogu P1642}/) 规划

题意:给一颗 /(n/) 个节点的树,每个节点都有权值 /(a_i,b_i/),要求在树上选出 /(n-m/) 个连通的节点,最大化 /(x=/dfrac{/sum a_i}{/sum b_i}/)

/(/rm Sol./)
题目所给的式子中含有 /(/dfrac{/sum a_i}{/sum b_i}/) 这一项,我们无法确定到底是选 /(a_i/) 更大的还是 /(b_i/) 更大的,这时
我们就必须要将 /(a_i,b_i/) 这两项化为一个整体求解。

/(/begin{aligned}x=/dfrac{/sum a_i}{/sum b_i}///to /sum a_i=x/cdot /sum b_i ///to/sum a_i-x/cdot/sum b_i =0///to /sum(a_i-xb_i)=0/end{aligned}/)

我们发现这样就可以将 /(a_i,b_i/) 两个参数化为一个处理。这题的 /(x/) 显然是有单调性的,我们二分 /(x/) 的值,将其带入上式,在树上跑树形 /(dp/) (背包)求其最大值,若最大值 /(/ge 0/) 说明当前 /(x/) 满足题意, /(ans = L = mid/),反之则 /(R = mid/)。

树形背包部分基本等于板子。

/(/rm Code:/)

#include <bits/stdc++.h>
using namespace std;
const int N = 105, M = 1e5 + 5;

#define db double 

db f[N][N], val[M], sum;
vector <int> Link[N];
db val1[M], val2[M];

int n, m;

void dp(int u, int Fa) {
    for(auto v : Link[u]) {
        if(v == Fa) continue;
        dp(v, u);
        for(int k = m; k; k--) 
            for(int j = k; j >= 0; j--) 
                f[u][k] = max(f[u][k], f[u][k - j] + f[v][j]); // 树形背包
    }
    for(int i = m; i >= 1; i--) f[u][i] = f[u][i - 1] + val[u]; // 为保证连通性强制取当前点
    sum = max(sum, f[u][m]); // 一边更新一遍取最大值
}

inline bool check(db x) {
    for(int i = 1; i <= n; i++) val[i] = val1[i] - x * val2[i]; // 统一计算权值
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            f[i][j] = -1e9; // 注意 f[i][0] = 0,不能全部 memset 为 -inf
    sum = -1.0;

    dp(1, 0);
    return sum >= 0;
}

signed main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("Ans.txt", "w", stdout);
#endif
    cin >> n >> m; m = n - m;
    for(int i = 1; i <= n; i++) cin >> val1[i];
    for(int i = 1; i <= n; i++) cin >> val2[i];
    for(int i = 1; i < n; i++) {
        static int x, y; cin >> x >> y;
        Link[x].push_back(y);
        Link[y].push_back(x);
    }

    db L = 0.0, R = 1e6, ans;
    while(L + 1e-5 < R) { // 二分答案
        db mid = (L + R) / 2;
        if(check(mid)) { ans = L = mid; }
        else R = mid;
    }
    printf("%.1lf", ans);
    return 0;
}

原创文章,作者:3628473679,如若转载,请注明出处:https://blog.ytso.com/273776.html

(0)
上一篇 2022年7月12日
下一篇 2022年7月12日

相关推荐

发表回复

登录后才能评论