Data update

Task/Partition-function-P/SETL/partition-function-p.setl

@@ -0,0 +1,36 @@
+program partition_function;
+    loop for n in [666, 6666] do
+        show_partition_with_time(n);
+    end loop;
+
+    proc show_partition_with_time(n);
+        s := clock;
+        p := partition(n);
+        d := (clock - s) / 1000;
+        print("p(" + str n + ") = " + str p + " (" + str d + "s)");
+    end proc;
+
+    proc partition(n);
+        pn := [1,1];
+        loop for i in [3..n+2] do
+            pn(i) := 0;
+            loop init k := 1; step k +:= 1; do
+                penta := k * (3 * k-1) div 2;
+                if penta >= i-1 then quit; end if;
+                if k mod 2 = 1 then
+                    pn(i) +:= pn(i-penta);
+                else
+                    pn(i) -:= pn(i-penta);
+                end if;
+                penta +:= k;
+                if penta >= i-1 then quit; end if;
+                if k mod 2 = 1 then
+                    pn(i) +:= pn(i-penta);
+                else
+                    pn(i) -:= pn(i-penta);
+                end if;
+            end loop;
+        end loop;
+        return pn(n+2);
+    end proc;
+end program;