题目链接:http://poj.org/problem?id=3111
题目大意:n个珠宝,每个珠宝对应一个重量wi和价值vi,现在要从中找出k个,使的这k个珠宝的单位重量的价值最大,输出这k个珠宝的编号。
分析:一般最先想到的方法可能是把物品按照单位价值进行排序,从大到小贪心的进行搜索,但这样做,实际求得的是对于给定的重量,可以得到的最大价值,显然这不是本题所要求的。
实际上,我们完全可以二分所有可能得到的价值,从中找出最大的价值。我们定义条件C(x)=可以选择使得单位重量的价值不小于x,这样原问题就变为了求满足C(x)的最大的x。至于怎么判断C(x)是否可行呢?假如我们选了某个物品的集合S,那么他的单位重量的价值是∑vi / ∑wi,因此就变为了判断是否存在S满足:∑vi / ∑wi >=x,化简可得∑(vi-x*wi)>=0,因此,我们可以对yi=(vi-x*wi)的值进行排序然后贪心选择,因此问题就变为了判断条件 C(x)=yi按降序排列的前k个的和不小于0 是否成立了。时间复杂度为O(n logn)。
实现代码如下:
#include <cstdio>
#include <algorithm>
using namespace std;
const int M=1e5+10;
const double inf=99999999.0;
struct node
{
int v,w,id;
double y;
bool operator <(const node &a) const
{
return y>a.y;
}
}jew[M];
int n,k;
bool judge(double x)
{
for(int i=0;i<n;i++)
jew[i].y=jew[i].v-x*jew[i].w;
sort(jew,jew+n);
double sum=0;
for(int i=0;i<k;i++)
sum+=jew[i].y;
return sum>=0;
}
void solve()
{
double left=0,right=inf;
for(int i=0;i<100;i++)
{
double mid=(left+right)/2;
if(judge(mid)) left=mid;
else right=mid;
}
for(int i=0;i<k-1;i++)
printf("%d ",jew[i].id);
printf("%d\n",jew[k-1].id);
}
int main()
{
while(scanf("%d%d",&n,&k)!=-1)
{
for(int i=0;i<n;i++)
{
scanf("%d%d",&jew[i].v,&jew[i].w);
jew[i].id=i+1;
}
solve();
}
return 0;
}