Mercurial > hg > problem6
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()