# HG changeset patch # User Jordi GutiƩrrez Hermoso # Date 1426023351 14400 # Node ID 15944d95f3995ab9bbfd50a13b9864e92ae89e69 # Parent 3ef3d3800092694c8ce2aea123415443e9240693 parseinput: new function diff --git a/optim.py b/optim.py 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()