Discrete : add 04-warehouse-location assignment

1 files changed, 75 insertions, 0 deletions

diff --git a/04-warehouse-location/solver.py b/04-warehouse-location/solver.py
new file mode 100644
index 0000000..823d7f0
--- /dev/null
+++ b/04-warehouse-location/solver.py
@@ -0,0 +1,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)'
+