diff options
Diffstat (limited to 'challenge-121/abigail/python/ch-2.py')
| -rw-r--r-- | challenge-121/abigail/python/ch-2.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/challenge-121/abigail/python/ch-2.py b/challenge-121/abigail/python/ch-2.py new file mode 100644 index 0000000000..c01e43bf43 --- /dev/null +++ b/challenge-121/abigail/python/ch-2.py @@ -0,0 +1,40 @@ +#!/opt/local/bin/python + +# +# See ../README.md +# + +# +# Run as: python ch-2.py < input-file +# + +import fileinput + +def shortest_path (matrix, start, to, exclude): + if 1 + len (exclude . keys ()) == len (matrix): + return matrix [start] [to], [to] + + shortest = 9999999999999999999 + sh_path = [] + new_exclude = exclude . copy () + new_exclude [start] = 1 + + for next in range (len (matrix)): + if next == start or next == to or next in exclude: + continue + length, path = shortest_path (matrix, next, to, new_exclude) + if shortest > length + matrix [start] [next]: + shortest = length + matrix [start] [next] + sh_path = [next] + path + + return shortest, sh_path + + +matrix = [] +for line in fileinput . input (): + matrix . append (list (map (lambda x: int (x), line . split (" ")))) + +length, path = shortest_path (matrix, 0, 0, {}) + +print (length) +print (0, " " . join (map (lambda x: str (x), path))) |
