Replace NQueens.cpp with NQueens.rb

OmeletWithoutEgg · Apr 6, 2024 · faef98e · faef98e
1 parent cd00a20
commit faef98e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/codes/Misc/NQueens.rb b/codes/Misc/NQueens.rb
@@ -0,0 +1,9 @@
+def solve(n)
+  if n % 6 == 2 then
+    (2..n).step(2) + [3,1] + (7..n).step(2) + [5]
+  elsif n % 6 == 3 then
+    (4..n).step(2) + [2] + (5..n).step(2) + [1,3]
+  else
+    (2..n).step(2) + (1..n).step(2)
+  end
+end
diff --git a/pdf/content.yaml b/pdf/content.yaml
@@ -431,7 +431,7 @@
       path: Josephus.cpp
       verified: 8137f19
     - name: N Queens Problem
-      path: NQueens.cpp
+      path: NQueens.rb
       verified:
     # - name: Tree Knapsack
     #   path: TreeKnapsack.cpp

diff --git a/pdf/gen_content.py b/pdf/gen_content.py
@@ -126,6 +126,11 @@ def gen_tex(sections: List[Dict[str, Any]], out: IO) -> None:
                     "\\IncludeTex{%s}{%s}\n"
                     % (escape_latex(content["name"]), content["path"])
                 )
+            elif ext == ".rb":
+                out.write(
+                    "\\IncludeCode[language=ruby]{%s}{%s}\n"
+                    % (escape_latex(content["name"]), content["path"])
+                )
             elif base == "vimrc":
                 out.write(
                     "\\IncludeCode[language=vim]{%s}{%s}\n"