POJ3111-K Best

最新推荐文章于 2019-08-03 10:14:31 发布

stdwal

最新推荐文章于 2019-08-03 10:14:31 发布

阅读量336

点赞数

CC 4.0 BY-SA版权

分类专栏： POJ 二分搜索

本文链接：https://blog.youkuaiyun.com/Walton_/article/details/52048716

POJ 同时被 2 个专栏收录

119 篇文章

订阅专栏

二分搜索

15 篇文章

订阅专栏

本文介绍了一种使用二分查找算法来最大化一组数据平均值的方法，并通过一个具体实例展示了如何实现这一算法。该方法适用于解决特定类型的问题，如资源分配等。

这道题与POJ2976类似，但是数据量貌似很大....

二分最大化平均值

#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

const int maxn = 100000;
const int inf = 10e6;

int v[maxn];
int w[maxn];

struct rec {
    double y;
    int id;
} r[maxn];

vector<int> vec;

bool comp(const rec& r1, const rec& r2) {
    return r1.y > r2.y;
}

bool judge(double x, int n, int k) {
    vec.clear();
    for (int i = 0; i < n; i++) {
        r[i].y = v[i] - x * w[i];
        r[i].id = i + 1;
    }
    sort(r, r + n, comp);
    double sum = 0;
    for (int i = 0; i < k; i++) {
        sum += r[i].y;
        vec.push_back(r[i].id);
    }
    return sum >= 0;
}

int main(int argc, char const *argv[]) {
    int n, k;
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i++) {
        scanf("%d%d", &v[i], &w[i]);
    }
    double lb = 0;
    double ub = inf;
    //50的精度已经够了，100就会超时...
    for (int i = 0; i < 50; i++) {
        double mid = (lb + ub) / 2;
        if (judge(mid, n, k)) {
            lb = mid;
        } else {
            ub = mid;
        }
    }
    for (int i = 0; i < vec.size(); i++) {
        if (i)
            putchar(' ');
        printf("%d", vec[i]);
    }
    putchar('\n');

    return 0;
}