ACM模版

描述

题解

机器调度问题,这是一个经典问题: 2台机器的情况下有多项式算法(Johnson算法),3台或以上的机器是NP-hard算法。

代码

#include <iostream>
#include <algorithm>
#include <cstdio>

using namespace std;

const int MAXN = 5e4 + 5;

struct task
{
    int a;
    int b;
} TaskA[MAXN], TaskB[MAXN];

bool cmpA(task a, task b)
{
    return a.a <= b.a;
}

bool cmpB(task a, task b)
{
    return a.b >= b.b;
}

int main(int argc, const char * argv[])
{
    int N;
    cin >> N;

    int a, b;
    int posA = 0, posB = 0;
    int sumA = 0, sumB = 0;
    for (int i = 0; i < N; i++)
    {
        scanf("%d %d", &a, &b);
        if (a < b)
        {
            TaskA[posA].a = a;
            TaskA[posA++].b = b;
            sumA += b;
        }
        else
        {
            TaskB[posB].a = a;
            TaskB[posB++].b = b;
            sumB += a;
        }
    }
    sort(TaskA, TaskA + posA, cmpA);
    sort(TaskB, TaskB + posB, cmpB);

    for (int i = 0; i < posB; i++)
    {
        TaskA[posA++] = TaskB[i];
    }

    int ans = TaskA[0].a + TaskA[0].b;
    int sum = TaskA[0].a;
    for (int i = 1; i < posA; i++)
    {
        sum += TaskA[i].a;
        ans = sum < ans ? ans + TaskA[i].b : sum + TaskA[i].b;
    }

    cout << ans << '\n';

    return 0;
}

参考

《机器工作调度》