01分数规划

01分数规划基本问题

给出两个有 $n$ 个元素的数组 $a_i$ 和 $b_i$，对于一组 w i ∈ { 0 , 1 } w_i\in \{0,1\} wi​∈{0,1}，可能使得 $\frac{\sum _{i=1}^n{a}_i{w}_i}{\sum _{i=1}^n{b}_i{w}_i}$ 取得最值，求这个最值。

求解——二分法

check函数

我们二分答案。假设要求的是最大值。
二分得到一个答案 $x$，假设当前 $x$ 小于最大值或者正好去到最大值，那么有 $\frac{\sum _{i=1}^n{a}_i{w}_i}{\sum _{i=1}^n{b}_i{w}_i}⩾x$，即 $\sum _{i=1}^n{w}_i\left(a_i-x{b}_i\right)⩾0$ 。如果经过检验，不存在$\sum _{i=1}^n{w}_i\left(a_i-x{b}_i\right)⩾0$ 的情况，即 {%raw%}$\max \left\{\sum _{i=1}^n{w}_i\left(a_i-x{b}_i\right)\right\}<0${%endraw%}，那么就与假设矛盾，根据反证法的思想，说明 $x$ 已经超过最大值。

模板

#include<iostream>
#include<cstdio>
#define N 100003
#define eps 1e-6
using namespace std;
double a[N],b[N];
inline bool check(double x)
{
	int i;
	for(i=1;i<=n;i++)
	{
		/*若存在a[i]-x*b[i]非负
		那么若干非负的a[i]-x*b[i]之和
		必然非负
		即x必然为可行值
		*/
		if(a[i]-x*b[i]>=0)
		{
			return 1;//返回真
		}
	}
	return 0;
}
int main()
{
	cin>>n;
	int i;
	for(i=1;i<=n;i++) scanf("%lf",a+i);
	for(i=1;i<=n;i++) scanf("%lf",b+i);
	double l=0,r=1e9;
	double ans=0;
	//二分
	while(r-l>=eps)
	{
		double mid=(l+r)/2;
		if(check(mid)) 
		{
			ans=l;//mid可行，更新最终答案
			l=mid;
		}
		else r=mid;//mid不可行，答案比mid小
	}
	printf("%.6lf\n",ans);
	return 0;
}

例题

牛客竞赛 NC15446 $\text{wyh}$ 的物品 $\looparrowright$

题目描述

$\text{wyh}$ 学长现在手里有 $n$ 个物品，这 $n$ 个物品的重量和价值都告诉你，然后现在让你从中选取 $k$ 个，问你在所有可能选取的方案中，最大的单位价值为多少（单位价值为选取的 $k$ 个物品的总价值和总重量的比值）。

输入描述

输入第一行一个整数 $T(1⩽T⩽10)$。
接下来有 $T$ 组测试数据，对于每组测试数据，第一行输入两个数 $n$ 和 $k$$(1⩽k⩽n⩽100000)$。
接下来有 $n$ 行，每行两个是 $a$ 和 $b$，代表这个物品的重量和价值。

输出描述

对于每组测试数据，输出对应答案，结果保留两位小数。

示例1

输入
1
3 2
2 2
5 3
2 1
输出
0.75

说明

对于样例来说，我们选择第一个物品和第三个物品，达到最优目的。

分析

设使得单位价值最大的 $k$ 个物品的索引为 $i_1,i_2,\cdots ,i_k$。当二分得到一个答案 $x$ ，假设 $x$ 小于最大单位价值或恰好为最大单位价值，那么有 $\sum _{j=1}^k\frac{a_{i_j}}{b_{i_j}}⩾x$，即 $\sum _{j=1}^k(a_{i_j}-x{b}_{i_j})⩾0$，若不存在 $\sum _{j=1}^k(a_{i_j}-x{b}_{i_j})⩾0$ 的方案，即$\max \left\{\sum _{j=1}^k(a_{i_j}-x{b}_{i_j})\right\}<0$，说明 $x$ 大于最大值，右边界减小，否则 $x$ 合法。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#define N 100003
#define eps 1e-6
using namespace std;
int value[N],weight[N];
double p[N];
int n,k;
bool check(double x)
{
	int i;
	for(i=1;i<=n;i++) p[i]=value[i]-x*weight[i];
	sort(p+1,p+1+n,greater<double>());
	double res=0;
	//前k大的value[i]-x*wight[i]之和即为最大值
	for(i=1;i<=k;i++) res+=p[i];
	return res>=0;//若非负则x合法
}
int main()
{
	int _;
	for(cin>>_;_;_--)
	{
		scanf("%d%d",&n,&k);
		int i;
		for(i=1;i<=n;i++) scanf("%d%d",weight+i,value+i);
		double l=0,r=1e9;
		double ans;
		//二分
		while(r-l>=eps)
		{
			double mid=(l+r)/2;
			if(check(mid))
			{
				ans=l;//mid合法，更新答案
				l=mid;
			}
			else r=mid;
		}
		printf("%.2lf\n",ans);
	}
	return 0;
}