From 5525015ff159ac4b55fcff3ab4501ad8f8857722 Mon Sep 17 00:00:00 2001 From: josch Date: Tue, 3 Jul 2012 22:16:45 +0200 Subject: [PATCH] read number of vertices and edges from commandline --- cycles.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/cycles.py b/cycles.py index 721eb2e..a439dad 100644 --- a/cycles.py +++ b/cycles.py @@ -52,18 +52,16 @@ procedure circuit_enumeration; end; """ -A = [[] for a in range(5)] +import sys -A[0].append(1) -A[0].append(4) -A[1].append(4) -A[2].append(0) -A[2].append(3) -A[2].append(4) -A[3].append(0) -A[3].append(2) -A[3].append(4) -A[4].append(3) +if len(sys.argv) < 3: + print "usage: %s num_vertices [v1,v2...]"%(sys.argv[0]) + +A = [[] for a in range(int(sys.argv[1]))] + +for edge in sys.argv[2:]: + v1,v2 = edge.split(',', 1) + A[int(v1)].append(int(v2)); def print_point_stack(): for p in point_stack: