零件分组【贪心算法习题python】

题目描述

某工厂生产一批棍状零件，每个零件都有一定的长度（Li）和重量（Wi）。现在为了加工需要，要将它们分成若干组，使每一组的零件都能排成一个长度和重量都不下降（若i<j，则Li<=Lj，Wi<=Wj）的序列。请问至少要分成几组?

输入输出格式

输入格式：

第一行为一个整数N（N<=1000），表示零件的个数。第二行有N对正整数，每对正整数表示这些零件的长度和重量，长度和重量均不超过10000。

输出格式：

仅一行，即最少分成的组数。

输入输出样例

输入样例#1：

5
8 4 3 8 2 3 9 7 3 5

输出样例#1：

2

分析：

1.优先按照长度排列，长度相等按照重量排列。

2.一次遍历零件，满足条件的零件加入同一组。

def min_group(num_obj, obj_feature_pair):
    pair_list = []      # 组成对
    processed_list = []     # 被分组过的零件
    group = 0   # 组数
    for i in range(0, len(obj_feature_pair), 2):
        pair_list.append([obj_feature_pair[i], obj_feature_pair[i+1]])
    pair_list.sort(key=lambda i:(i[0],i[1]))    # 按照长度-》重量排列零件
    while len(processed_list) != num_obj:
        min_f1 = -float('inf')
        min_f2 = -float('inf')
        start_flag = False    # 出现满足条件的零件标识
        for index, pair in enumerate(pair_list):
            if pair[0] >= min_f1 and pair[1] >= min_f2 and index not in processed_list:
                start_flag = True
                min_f1 = pair[0]
                min_f2 = pair[1]
                processed_list.append(index)
        if start_flag:
            group += 1
    return group

print(min_group(5, [8, 4, 3, 8, 2, 3, 9, 7, 3, 5]))