blob: 823d7f0704096d8d3bdbf0f65370aba66f5e67d2 (
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
def solveIt(inputData):
# Modify this code to run your optimization algorithm
# parse the input
lines = inputData.split('\n')
parts = lines[0].split()
warehouseCount = int(parts[0])
customerCount = int(parts[1])
warehouses = []
for i in range(1, warehouseCount+1):
line = lines[i]
parts = line.split()
warehouses.append((int(parts[0]), float(parts[1])))
customerSizes = []
customerCosts = []
lineIndex = warehouseCount+1
for i in range(0, customerCount):
customerSize = int(lines[lineIndex+2*i])
customerCost = map(float, lines[lineIndex+2*i+1].split())
customerSizes.append(customerSize)
customerCosts.append(customerCost)
# build a trivial solution
# pack the warehouses one by one until all the customers are served
solution = [-1] * customerCount
capacityRemaining = [w[0] for w in warehouses]
warehouseIndex = 0
for c in range(0, customerCount):
if capacityRemaining[warehouseIndex] >= customerSizes[c]:
solution[c] = warehouseIndex
capacityRemaining[warehouseIndex] -= customerSizes[c]
else:
warehouseIndex += 1
assert capacityRemaining[warehouseIndex] >= customerSizes[c]
solution[c] = warehouseIndex
capacityRemaining[warehouseIndex] -= customerSizes[c]
used = [0]*warehouseCount
for wa in solution:
used[wa] = 1
# calculate the cost of the solution
obj = sum([warehouses[x][1]*used[x] for x in range(0,warehouseCount)])
for c in range(0, customerCount):
obj += customerCosts[c][solution[c]]
# prepare the solution in the specified output format
outputData = str(obj) + ' ' + str(0) + '\n'
outputData += ' '.join(map(str, solution))
return outputData
import sys
if __name__ == '__main__':
if len(sys.argv) > 1:
fileLocation = sys.argv[1].strip()
inputDataFile = open(fileLocation, 'r')
inputData = ''.join(inputDataFile.readlines())
inputDataFile.close()
print 'Solving:', fileLocation
print solveIt(inputData)
else:
print 'This test requires an input file. Please select one from the data directory. (i.e. python solver.py ./data/wl_16_1)'
|
