题目大意:.,这样的一个公式,给出N对vi,wi,选择其中的K对,使得改公式的和最大。
仍然是个二分搜索的题,设c(x):=s不小于x。只要对vi-x*wi进行排序,然后选择最大的k个加起来,如果大于0则命题成立。需要注意的是精度问题,1e-6这个精度是可以AC的。
#include <stdio.h>
#include <vector>
#include <math.h>
#include <string.h>
#include <string>
#include <iostream>
#include <queue>
#include <list>
#include <algorithm>
#include <stack>
#include <map>
using namespace std;
struct SCORE
{
int ivalue;
int bvalue;
int number;
};
SCORE values[100001];
double X = 0;
int compp(const void* a1, const void* a2)
{
SCORE* score1 = (SCORE*)a1;
SCORE* score2 = (SCORE*)a2;
double dis = score2->ivalue - score2->bvalue * X - (score1->ivalue - score1->bvalue * X);
if (dis > 0)
{
return 1;
}
else if (dis < 0)
{
return -1;
}
else
return 0;
}
bool CC(double x, int n, int k)
{
int select = k;
X = x;
qsort(values, n, sizeof(SCORE), compp);
long double sum = 0;
for (int i = 0; i < select; i++)
{
sum += (values[i].ivalue - values[i].bvalue * x);
}
if (sum - 0.0 >= 0)
{
return true;
}
else
return false;
}
int main()
{
#ifdef _DEBUG
freopen("e:\\in.txt", "r", stdin);
#endif
int n, k;
scanf("%d %d", &n, &k);
int maxsum = 0;
for (int i = 0; i < n; i++)
{
scanf("%d %d\n", &values[i].ivalue, &values[i].bvalue);
values[i].number = i;
maxsum += values[i].ivalue;
}
double l = 0;
double r = maxsum + 1;
int count = 0;
while (fabs(r - l) > 1e-6)
{
count++;
double mid = (l + r) / 2;
if (CC(mid, n, k))
{
l = mid;
}
else
r = mid;
}
for (int i = 0; i < k; i++)
{
printf("%d ",values[i].number + 1);
}
return 1;
}
此题还有一种更快的方法,就是用迭代法求,初始时令s = 最大的k个vi - wi的和,然后将s代入 s1 = 最大的vi - s*wi的和,一直迭代到s的值变化少于1e-6即可。这个方法是POJ某位大神的超快解法。