blob: e2b8af772546d038c2e1fae49dce76eeb7b17c68 (
plain)
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
|
#!/usr/bin/python3
file = open("../input.txt", "r")
story_table = []
stories_table = []
nb_machines = 0
loop_number = 0
for line in file:
line_table = list(map(int, line.split()))
keys_line_table_story = ['machines', 'money', 'time']
keys_line_table_machines = ['day_sell', 'buy_price', 'resale_price', 'day_profit']
l = list()
for a in range(0, len(line_table)):
l.append((line_table[a]))
if len(line_table) == 3:
line_table = dict(zip(keys_line_table_story,l))
else:
line_table = dict(zip(keys_line_table_machines,l))
if len(line_table) == 3:
nb_machines = line_table["machines"]
if loop_number > 0:
stories_table.append(story_table)
story_table = []
story_table.append(line_table)
loop_number += 1
case = 1
for story in stories_table:
settings = story[0]
current_money = settings["money"]
machines_bought = []
for day in range(0, settings["time"]+1):
for machines in machines_bought:
current_money += machines["day_profit"]
buyable_machines = []
for machine_number in range(1, settings["machines"]+1):
if day == story[machine_number]["day_sell"]:
buyable_machines.append(story[machine_number])
for machines in buyable_machines:
if machines["buy_price"] <= current_money:
machines_bought.append(machines)
current_money -= machines["buy_price"]
print("Case "+ str(case) + ": " + str(current_money))
case += 1
|
