Mercurial > hg > problem6
comparison optim.py @ 1:15944d95f399 draft
parseinput: new function
author | Jordi GutiƩrrez Hermoso <jordigh@octave.org> |
---|---|
date | Tue, 10 Mar 2015 17:35:51 -0400 (2015-03-10) |
parents | |
children | 3632502b8af1 |
comparison
equal
deleted
inserted
replaced
0:3ef3d3800092 | 1:15944d95f399 |
---|---|
1 #!/usr/bin/env python | |
2 | |
3 def parseinput(fname): | |
4 """ | |
5 Parse the input file, forget about input validation | |
6 """ | |
7 cases = [] | |
8 with open(fname) as f: | |
9 while True: | |
10 header = [int(x) for x in f.readline().split()] | |
11 if header == [0, 0, 0]: | |
12 return cases | |
13 N = header[0] | |
14 case = {"header": header, | |
15 "machines": []} | |
16 for i in range(0, N): | |
17 machine = [int(x) for x in f.readline().split()] | |
18 case["machines"].append(machine) | |
19 cases.append(case) | |
20 | |
21 def main(): | |
22 cases = parseinput("input.txt") | |
23 for case in cases: | |
24 print "Next case:", case["header"] | |
25 for machine in case["machines"]: | |
26 print machine | |
27 | |
28 if __name__ == "__main__": | |
29 main() |