K Best [二分]

最新推荐文章于 2021-03-28 23:38:35 发布

原创最新推荐文章于 2021-03-28 23:38:35 发布 · 460 阅读

0 ·

CC 4.0 BY-SA版权

二分专栏收录该内容

9 篇文章

订阅专栏

本文提供了一个使用C++进行编程竞赛的示例代码。该示例涉及输入输出操作、结构体定义及排序等基本算法知识，通过双指针法找到满足条件的最优解，并输出结果。

这题与http://blog.youkuaiyun.com/qer_computerscience/article/details/77601173
一模一样。

#include<iostream>
#include<string>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<cstdlib>
#include<list>
#include<stack>
#include<cmath>
#include<iomanip>
using namespace std;
//#pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
void debug() {cout << "ok running!" << endl;}
int n, k;
struct node
{
    int v, w, id;
    double x;
    bool operator < (node& b)
    {
        return x > b.x;
    }
} a[100005];
bool check(double mid)
{
    for(int i = 0; i < n; ++i)
        a[i].x = a[i].v - mid*a[i].w;
    sort(a, a+n);
    double sum = 0;
    for(int i = 0; i < k; ++i)
        sum += a[i].x;
    if(sum >= 0)
        return 1;
    else return 0;
}
int main()
{
    ios::sync_with_stdio(false);
    #ifndef ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
    #endif // ONLINE_JUDGE
    while(cin >> n >> k)
    {
        for(int i = 0; i < n; ++i)
        {
            int x, y;
            cin >> x >> y;
            a[i].id = i+1;
            a[i].v = x, a[i].w = y;
        }
        double l = 0.0, r = 999999999.0;
        while(r - l > 1e-4)
        {
            double mid = (l+r) / 2;
            //cout << mid << endl;
            if(check(mid))
                l = mid;
            else r = mid;
        }
        for(int i = 0; i < k-1; ++i)
            cout << a[i].id << " ";
        cout << a[k-1].id << endl;
    }
    return 0;
}