【贪心】加工生产调度

问题 D: 【贪心】加工生产调度
时间限制: 1 Sec 内存限制: 64 MB
[提交] [状态]
题目描述
有n个部件需在A、B机器上加工，每个工件都必须经过先A后B两道工序。
已知：部件i在A、B机器上的加工时间分别为ai，bi。
问：如何安排n个工件的加工顺序，才能使得总加工时间最短？
输入
第1行仅一个整数n (0<n<1000)，表示产品的数量；
第2行n个整数，表示这n个产品在A车间加工各自所要的时间（都是整数）；
第3行n个整数，表示这n个产品在B车间加工各自所要的时间（都是整数）。
输出
只有一个数，表示最少的加工时间；
样例输入 Copy
5
3 5 8 7 10
6 2 1 4 9
样例输出 Copy
34

我的代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
 
using namespace std;
 
struct node{int a,b,flag;}item[1010];
bool cmp(node x, node y)
{
    if(x.flag == y.flag)
    {
        if(x.flag == 0)
            return x.a < y.a;
        return x.b>y.b;
    }
    return x.flag<y.flag;
}
 
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0; i<n; i++)
        scanf("%d",&item[i].a);
    for(int i=0; i<n; i++)
        scanf("%d", &item[i].b);
    for(int i=0; i<n; i++)
        item[i].flag = (item[i].a < item[i].b?0:1);
    sort(item,item+n,cmp);
    int A = 0, B = 0;
    for(int i=0; i<n; i++)
    {
        A+=item[i].a;
        B = max(B,A) + item[i].b;
    }
    printf("%d\n",B);
    return 0;
}