最大化平均值(二分搜索)

有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);
}






评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值