有n个物品的重量和价值分别是w[i]和v[i]。从中选出k个物品使得单位重量的价值最大。
限制条件:
1<=k<=n<=10^4
1<=w[i],v[i]<=10^6
输入
n=3
k=2
(w,v)={(2,2),(5,3),(2,1)}
输出
0.75(选0号和2号物品,(2+1)/(2+2)=0.75)
思路:一般最先想到的方法可能是吧物品按照单位价值进行排序,从大到小贪心地进行选取。但是这个方法对于样例输入得到的结果是5/7=0.714。所以这个方法是不可行的;
实际上,对于这个问题使用二分搜索法可以很好地解决。我们定义:
C(x):=可以选择使得单位重量的价值不小于x
因此,原问题就变成了求满足C(x)的最大的x。假设我们选了某个物品的集合S,那么它们的单位重量的价值是:∑v[i]/∑w[i](i∈S),因此就变成了判断是否存在S满足下面的条件∑v[i]/∑w[i](i∈S)>=x,把这个不等式进行变形就得到∑(v[i]-x*w[i])>=0,因此,可以对(v[i]-x*w[i])的值进行排序贪心地进行选取。因此就变成了
C(x)=(v[i]-x*w[i])从大到小排列中的前k个的和不小于0
每次判断的复杂度是O(nlogn);
代码:
//输入
int n,k;
int w[maxn],v[maxn];
double y[maxn];//v-x*w
//判断是否满足条件
bool C(double x){
for(int i=0;i<n;i++)
y[i]=v[i]-x*w[i];
sort(y,y+n);
//计算y数组中从大到小前k个数的和
double sum=0;
for(int i=0;i<k;i++)
sum+=y[n-i-1];
return sum>=0;
}
void solve(){
double lb=0;ub=INF;
for(int i=0;i<100;i++){
double mid=(lb+ub)/2;
if(C(mid)) lb=mid;
else ub=mid;
}
printf("%.2f\n",ub);
}