题目描述:
Demy有n颗钻石,每颗钻石都有价值和质量,现在要带走其中的k颗钻石,要使得单位价值最大,即Σv/Σw的值最大。
大致思路:
这个题很明显就是要最大化平均值,然而采用贪心的方法每次取单位价值最大的钻石,是显然不行的。所以应该采用二分搜索的方法,那么二分什么值最后才能得到答案呢?不妨这样想,S(x)表示最终的平均价值,那么就相当于找到一组(v,w)组合使得Σv/Σw≥S(x),移项得到不等式Σ(v-S(x)*w)≥0,这样一来就很容易发现直接二分S(x)即可,每次得到一个S(x)计算v-S(x)*w,之后排序,取前k个计算是否大于等于0,知道二分到一个S(x)使得不等式的值为0就是答案。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <string>
#include <map>
#include <queue>
#include <set>
using namespace std;
const int maxn = 100000 + 5;
const double inf = 1000000 + 5;
int n,k;
struct jew {
int id;
int v,w;
double key;
void cal(double x) { key = v - x * w; }
bool operator < (const jew& a) const {
return key > a.key;
}
}j[maxn];
bool c(double x) {
for (int i = 1; i <= n; i++) {
j[i].cal(x);
}
sort(j+1,j+n+1);
double tmp = 0;
for (int i = 1; i <= k; i++) tmp += j[i].key;
return tmp >= 0;
}
int main() {
while (cin>>n>>k) {
for (int i = 1; i <= n; i++) {
scanf("%d%d",&j[i].v,&j[i].w);
j[i].id = i;
}
double l = 0, r = inf;
for (int i = 0; i < 100; i++) {
double mid = (l + r) / 2;
if (c(mid)) l = mid;
else r = mid;
}
for (int i = 1; i <= k; i++) {
if (i - 1) printf(" %d",j[i].id);
else printf("%d",j[i].id);
}
cout<<endl;
}
}