changeset 1:15944d95f399 draft

parseinput: new function
author Jordi Gutiérrez Hermoso <jordigh@octave.org>
date Tue, 10 Mar 2015 17:35:51 -0400
parents 3ef3d3800092
children 3632502b8af1
files optim.py
diffstat 1 files changed, 29 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
new file mode 100755
--- /dev/null
+++ b/optim.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+
+def parseinput(fname):
+    """
+    Parse the input file, forget about input validation
+    """
+    cases = []
+    with open(fname) as f:
+        while True:
+            header = [int(x) for x in f.readline().split()]
+            if header == [0, 0, 0]:
+                return cases
+            N = header[0]
+            case = {"header": header,
+                    "machines": []}
+            for i in range(0, N):
+                machine = [int(x) for x in f.readline().split()]
+                case["machines"].append(machine)
+            cases.append(case)
+
+def main():
+    cases = parseinput("input.txt")
+    for case in cases:
+        print "Next case:", case["header"]
+        for machine in case["machines"]:
+            print machine
+
+if __name__ == "__main__":
+    main()