blob: 4685feead8031aa5e4d5930cc2761872bfdfc925 (
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
|
#!/bin/env python3
def tanken(drives, cost):
passagers = {}
distance = 0.
for d in drives:
try:
distance += int(d[0])
except:
return None, "Lines have to start with the driven distance!"
for p in d[1:]:
if p not in passagers:
passagers[p] = [int(d[0]),0]
else:
passagers[p][0] += int(d[0])
c = cost/distance
for d in drives:
c_d = c * int(d[0]) / (len(d) - 1)
for p in d[1:]:
passagers[p][1] += c_d
return passagers, None
|