Newcoder 143 A.gpa（01分数规划+二分）

最新推荐文章于 2023-02-07 15:58:34 发布

v5zsq

最新推荐文章于 2023-02-07 15:58:34 发布

阅读量197

点赞数

分类专栏： Newcoder 二分三分

本文链接：https://blog.csdn.net/V5ZSQ/article/details/82748609

版权

Newcoder 同时被 2 个专栏收录

186 篇文章 0 订阅

订阅专栏

二分三分

81 篇文章 0 订阅

订阅专栏

Description

有 $n$ 门课程，第 $i$ 门学分为 $s_i$ ，得分为 $c_i$ ，所选课程的总 $g p a$ 为 $\frac{\sum s_ic_i}{\sum s_i}$ ，问删除 $k$ 门课程后剩余课程 $g p a$ 的最大值

Input

第一行两个整数 $n, k$ ，之后输入 $n$ 个整数 $s_i$ ，最后输入 $n$ 个整数 $c_i$

$(1\le n\le 10^5,0\le k<n,1\le s_i,c_i\le 10^3)$

Output

输出最大 $g p a$

Sample Input

3 1
1 2 3
3 2 1

Sample Output

2.33333333333

Solution

$01$ 分数规划，二分答案 $x$ ，那么问题转化为删去 $k$ 个元素使得 $\sum (s_ic_i-xs_i)\ge 0$ ，排完序选最小的 $k$ 个删掉即可

Code

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define maxn 100005
#define eps 1e-8
int a[maxn],b[maxn];
double d[maxn];
int n,k;
int sgn(double x)
{
	if(fabs(x)<eps)return 0;
	if(x>eps)return 1;
	return -1;
}
double C(double x)
{
    double ans=0;
    for(int i=0;i<n;i++)d[i]=1.0*a[i]-x*b[i],ans+=d[i];
    sort(d,d+n);
    for(int i=0;i<k;i++)ans-=d[i];
    return sgn(ans)>=0;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0;i<n;i++)scanf("%d",&b[i]);
        for(int i=0;i<n;i++)scanf("%d",&a[i]),a[i]*=b[i];
        double l=0,r=1000,ans=-1;
        int T=30;
        while(T--)
        {
            double mid=0.5*(l+r);
            if(C(mid))l=ans=mid;
            else r=mid;
        }
        printf("%.6f\n",ans);
    }
    return 0;
}