aboutsummaryrefslogtreecommitdiff
path: root/challenge-115/abigail/lua/ch-1.lua
diff options
context:
space:
mode:
authorAbigail <abigail@abigail.be>2021-06-01 19:36:31 +0200
committerAbigail <abigail@abigail.be>2021-06-04 01:44:15 +0200
commit2fed574178675ba5e0cf2519e2696340ee608eed (patch)
tree78e036f5a802e08d52ef51b67762d5228e2023ab /challenge-115/abigail/lua/ch-1.lua
parent4f04dab2323d4219a5eb906c4bbc28059fecb10c (diff)
downloadperlweeklychallenge-club-2fed574178675ba5e0cf2519e2696340ee608eed.tar.gz
perlweeklychallenge-club-2fed574178675ba5e0cf2519e2696340ee608eed.tar.bz2
perlweeklychallenge-club-2fed574178675ba5e0cf2519e2696340ee608eed.zip
AWK, Bash, C, Lua, Node.js, Perl, Python and Ruby solutions for week 115, part 1
Diffstat (limited to 'challenge-115/abigail/lua/ch-1.lua')
-rw-r--r--challenge-115/abigail/lua/ch-1.lua64
1 files changed, 64 insertions, 0 deletions
diff --git a/challenge-115/abigail/lua/ch-1.lua b/challenge-115/abigail/lua/ch-1.lua
new file mode 100644
index 0000000000..650c2e1c1b
--- /dev/null
+++ b/challenge-115/abigail/lua/ch-1.lua
@@ -0,0 +1,64 @@
+#!/opt/local/bin/lua
+
+--
+-- See ../README.md
+--
+
+--
+-- Run as: lua ch-1.lua < input-file
+--
+
+for line in io . lines () do
+ local graph = {}
+ local nodes = {}
+ for s in line : gmatch ("%S+")
+ do local first = s : sub ( 1, 1)
+ local last = s : sub (-1, -1)
+ if graph [first] == nil
+ then graph [first] = {}
+ end
+ graph [first] [last] = 1
+ nodes [first] = 1
+ nodes [last] = 1
+ end
+
+ --
+ -- Make sure all entries exists, as lua doesn't autovivify
+ --
+ for node1 in pairs (nodes)
+ do for node2 in pairs (nodes)
+ do if graph [node1] == nil
+ then graph [node1] = {}
+ end
+ if graph [node1] [node2] == nil
+ then graph [node1] [node2] = 0
+ end
+ end
+ end
+
+ --
+ -- Calculate the transitive closure
+ --
+ for k in pairs (nodes)
+ do for i in pairs (nodes)
+ do for j in pairs (nodes)
+ do if graph [i] [j] == 0 and graph [k] [j] == 1 and
+ graph [i] [k] == 1
+ then graph [i] [j] = 1
+ end
+ end
+ end
+ end
+
+ --
+ -- We have a loop iff there is a node which is reachable from itself
+ --
+ local out = 0
+ for i in pairs (nodes)
+ do if graph [i] [i] == 1
+ then out = 1
+ end
+ end
+
+ print (out)
+end