Coconut week 122 task 2

1 files changed, 17 insertions, 0 deletions

diff --git a/challenge-122/luc65r/coconut/ch-2.coco b/challenge-122/luc65r/coconut/ch-2.coco
new file mode 100755
index 0000000000..081ef09edf
--- /dev/null
+++ b/challenge-122/luc65r/coconut/ch-2.coco
@@ -0,0 +1,17 @@
+#!/usr/bin/env -S coconut-run --strict
+
+import sys
+
+flatten = reduce $ (::)
+
+def scores(0) = [[]]
+
+@addpattern(scores)
+def scores(n) =
+    range(1, 4) ↦ filter $ ((≥) $ n) ↦ map $ (
+        i → scores(n − i) ↦ map $ (→ [i] + _)
+    ) ↦ flatten
+
+sys.argv[1] ↦ int ↦ scores ↦ map $ (
+    l → ' '.join(map(str, l))
+) ↦ '\n'.join ↦ print