-
Notifications
You must be signed in to change notification settings - Fork 1
/
new.py
116 lines (104 loc) · 4.36 KB
/
new.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import sys
inp_file = open(sys.argv[1], 'r')
class Contributer:
def __init__(self, name, skills_level: dict):
self.name = name
self.skills_level = skills_level
self.available = 0
def make_contribute(self,time:int):
self.available = time
def update_round(self):
self.available -= 1
class Project:
def __init__(self, name, skills_level: dict, time:int, score:int, best_before:int, no_of_contributers:int):
self.name = name
self.skills_level = skills_level
self.time = time
self.best_before = best_before
self.no_of_contributers = no_of_contributers
no_of_contributers, no_of_projects = map(int, inp_file.readline().split())
contributers = []
for i in range(no_of_contributers):
contributer_line = inp_file.readline().split()
contributer_name = contributer_line[0]
skills_level = {}
contributer_skills = int(contributer_line[1])
for j in range(contributer_skills):
contributer_skill = inp_file.readline().split()
contributer_skill_name = contributer_skill[0]
contributer_skill_level = int(contributer_skill[1])
skills_level[contributer_skill_name] = contributer_skill_level
contributer = Contributer(contributer_name, skills_level)
contributers.append(contributer)
projects = []
for i in range(no_of_projects):
project_line = inp_file.readline().split()
project_name = project_line[0]
skills_level = {} #{skill,level}
project_skills = int(project_line[-1])
for j in range(project_skills):
project_skill = inp_file.readline().split()
project_skill_name = project_skill[0]
project_skill_level = int(project_skill[1])
skills_level[project_skill_name] = project_skill_level
project = Project(project_name, skills_level, int(project_line[1]), int(project_line[2]), int(project_line[3]),int(project_line[4]))
#(name , skill , duration{time} , score , best_before , no. of contributors)
projects.append(project)
## Nubs First
# find minimum skill required of all projects
min_skill_required = {}
for project in projects:
for skill in project.skills_level:
if skill not in min_skill_required:
min_skill_required[skill] = [project.skills_level[skill],project,skill]
else:
if min_skill_required[skill][0] > project.skills_level[skill]:
min_skill_required[skill] = [project.skills_level[skill],project,skill]
# project order by skill
project_order_by_skill = []
for skill in min_skill_required:
project_order_by_skill.append(min_skill_required[skill])
project_order_by_skill.sort(key=lambda x: x[0])
stop = False
ans = []
while not stop:
# allot contributers to projects
stop = True
for project in project_order_by_skill:
is_valid_project = True
min_contributer = None
for contributer in contributers:
if contributer.available == 0:
if min_contributer == None:
min_contributer = contributer
if project[2] in min_contributer.skills_level and project[2] in contributer.skills_level and min_contributer.skills_level[project[2]] > contributer.skills_level[project[2]]:
min_contributer = contributer
if min_contributer:
# check project's skill level required
contri_to_project = [min_contributer]
for skill in project[1].skills_level:
if is_valid_project:
found = False
for con in contributers:
if con not in contri_to_project:
if con.skills_level[skill] >= project[1].skills_level[skill]:
contri_to_project.append(con)
found = True
if not found:
is_valid_project = False
break
if is_valid_project:
ans.append(project[1],contri_to_project)
stop = False
for con in contri_to_project:
con.make_contribute(project[1].time)
con.update_round()
if stop:
break
# print answer
print(len(ans))
for a in ans:
print(a[0].name)
for con in a[1]:
print(con.name,end=" ")
print()