from main import * from output import Output output = Output() pizzas.sort(key=lambda p: -len(p[1])) current_pizza = 0 for team_size, team_number in [(2, T2), (3, T3), (4, T4)]: for team in range(team_number): if current_pizza + team_size - 1 < len(pizzas): output.add_team( list( map(lambda x: pizzas[x][0], range(current_pizza, current_pizza + team_size)))) current_pizza += team_size else: break print("Score: " + str(output.score())) output.write(FILE_NAME)
best_id = -1 best_size = 0 for i in range(len(teams)): if len(teams[i][0]) >= team_size_by_id(i): continue new_ingredients_number = len(teams[i][1].union(pizza[1])) #if len(teams[i][1].union(pizza[1])) == len(pizza[1]) + len(teams[i][1]): # teams[i][0].append(pizza[0]) # teams[i][1] = teams[i][1].union(pizza[1]) # # best_id = -1 # # break if new_ingredients_number >= best_size: best_id = i best_size = new_ingredients_number if best_id != -1: teams[best_id][0].append(pizza[0]) teams[best_id][1] = teams[best_id][1].union(pizza[1]) #print(teams) for i in range(len(teams)): if team_size_by_id(i) == len(teams[i][0]): output.add_team(teams[i][0]) print("Score: " + str(output.score())) output.write(FILE_NAME)