#10003 「一本通 1.1 例 4」加工生产调度

解题思路

不妨把所有数据分为以下两类：第一类，工件在A车间加工时长小于等于B车间；第二类，工件在A车间加工时长大于B车间。如果对第一类零件进行生产安排，由于任意零件在A车间加工时长大于B车间，无论怎么安排，第一类零件在A车间加工总时间总是大于B车间；相似地，第二类零件在B车间加工总时间大于A车间，由此，可以得出，如果先安排第一类零件，再安排第二类零件，就可以取长补短，实现最短加工时长。对于大类内部的排序，由画图得，一类零件按A车间加工时间升序排列，二类零件按B车间加工时间降序排列。

完整代码

#include <iostream>
#include <algorithm>

using namespace std;

struct TIME
{
    int a;
    int b;
    int index;
};

bool cmp_a(const TIME& task1, const TIME& task2)
{
    return(task1.a < task2.a);
}

bool cmp_b(const TIME& task1, const TIME& task2)
{
    return(task1.b > task2.b);
}


int main()
{
    int n;
    cin >> n;

    TIME task[1005] = {0};
    for(int i = 0; i < n; i++)
    {
        cin >> task[i].a;
        task[i].index = i+1;
    }

    for(int i = 0; i < n; i++)
        cin >> task[i].b;

    TIME A[1005];
    TIME B[1005];

    int ia = 0;                      //一类零件总数
    int ib = 0;                      //二类零件总数
    int sum_a = 0;                   //一类零件A车间加工总时间 
    int sum_A = 0;                   //二类零件A车间加工总时间  
    int sum_b = 0;                   //一类零件B车间加工总时间  
    int sum_B = 0;                   //二类零件B车间加工总时间  
    for(int i = 0; i < n; i++)        
    {
        if(task[i].a <= task[i].b)
        {
            A[ia] = task[i];
            sum_a += A[ia].a;
            sum_B += A[ia].b;
            ia++;
        }
        else
        {
            B[ib] = task[i];
            sum_b += B[ib].b;
            sum_A += B[ib].a;
            ib++;
        }
    }

    sort(A, A+ia, cmp_a);
    sort(B, B+ib, cmp_b);

    sum_B += A->a;            //一类在B车间总时间需要增加在一类一项在A车间加工时间
    sum_A += (B+ib-1)->b;     //二类在A车间总时间需要增加在二类最后一项在A车间加工时间

    int result = max(sum_a+sum_A, sum_B+sum_b);

    cout << result << endl;
    for(int i = 0; i < ia; i++)
        cout << A[i].index <<" ";
    for(int i = 0; i < ib-1; i++)
        cout << B[i].index <<" ";
    cout << B[ib-1].index;
    return 0;
}