1. 程式人生 > >0/1 分數規劃

0/1 分數規劃

strong off splay += while inline 代碼 space tdi

模型:

給定整數 \(v_i, c_i\),規定 \(x_i=0\)\(1\),存在一組解 \(\{x_i\}\),使得 \(\displaystyle \frac{\sum_{i=1}^{n} v_ix_i}{\sum_{i=1}^{n} c_ix_i}\) 最大。

解法:

最大化 \(\displaystyle \frac{v_i}{c_i}\)(即性價比)的貪心方法不可行。

\(\displaystyle \frac{\sum_{i=1}^{n} v_ix_i}{\sum_{i=1}^{n} c_ix_i}\ge R\) 變式為 \(\sum_{i=1}^{n} (v_i-R\cdot c_i)x_i\ge 0\)

二分答案 \(R\),對於 \(R(\text{mid})\),計算 \(\sum_{i=1}^{n} (v_i-R\cdot c_i)x_i\) 的最大值,若最大值非負,令 \(l=\text{mid}\) (\(R\) 偏小),否則 \(r=\text{mid}\) (\(R\) 偏大)。

代碼:

(ssoj2388 coffee)

#include <cstdio>
#include <algorithm>
#define db double
#define eps 1e-5
using namespace std;

int n, m;
struct node {
    int w, c; db r;
    bool operator < (const node& A) const {return r>A.r; }
} G[203];

int main() {
    scanf("%d%d", &n, &m);
    for (int i=1; i<=n; ++i) scanf("%d", &G[i].w);
    for (int i=1; i<=n; ++i) scanf("%d", &G[i].c);
    db l=0.0, r=1000.0;
    while (l+eps<r) {
        db mid=(l+r)/2.0;
        for (int i=1; i<=n; ++i) G[i].r=G[i].w-mid*G[i].c;
        sort(G+1, G+n+1);
        db sum=0.0;
        for (int i=1; i<=m; ++i) sum+=G[i].r;
        if (sum<0) r=mid; else l=mid;
    }
    printf("%.3lf\n", l);
    return 0;
}

0/1 分數規劃