Add nth-prime

Author: keiravillekode

config.json

@@ -303,6 +303,14 @@
         "prerequisites": [],
         "difficulty": 6
       },
+      {
+        "slug": "nth-prime",
+        "name": "Nth Prime",
+        "uuid": "ba707f5a-61ba-45dd-a5bb-b2b573753f91",
+        "practices": [],
+        "prerequisites": [],
+        "difficulty": 6
+      },
       {
         "slug": "pascals-triangle",
         "name": "Pascal's Triangle",

exercises/practice/nth-prime/.docs/instructions.md

@@ -0,0 +1,7 @@
+# Instructions
+
+Given a number n, determine what the nth prime is.
+
+By listing the first six prime numbers: 2, 3, 5, 7, 11, and 13, we can see that the 6th prime is 13.
+
+If your language provides methods in the standard library to deal with prime numbers, pretend they don't exist and implement them yourself.

exercises/practice/nth-prime/.meta/config.json

@@ -0,0 +1,19 @@
+{
+  "authors": [
+    "keiravillekode"
+  ],
+  "files": {
+    "solution": [
+      "nth_prime.s"
+    ],
+    "test": [
+      "nth_prime_test.c"
+    ],
+    "example": [
+      ".meta/example.s"
+    ]
+  },
+  "blurb": "Given a number n, determine what the nth prime is.",
+  "source": "A variation on Problem 7 at Project Euler",
+  "source_url": "https://projecteuler.net/problem=7"
+}

exercises/practice/nth-prime/.meta/example.s

@@ -0,0 +1,46 @@
+.text
+.globl prime
+
+/*
+| Register | Usage      | Type    | Description             |
+| -------- | ---------- | ------- | ----------------------- |
+| `a0`     | output     | integer | prime                   |
+| `a0`     | input      | integer | remaining primes        |
+| `a1`     | temporary  | integer | 6                       |
+| `t0`     | temporary  | integer | prime candidate         |
+| `t1`     | temporary  | integer | prime step              |
+| `t2`     | temporary  | integer | factor candidate        |
+| `t3`     | temporary  | integer | factor step             |
+*/
+
+/* extern unsigned prime(unsigned number); */
+prime:
+        li      a1, 6
+        li      t1, 4                   /* prime step */
+        li      t0, 1                   /* prime candidate will be 5 */
+        addi    a0, a0, -2
+        bgtz    a0, outer
+
+        addi    a0, a0, 3               /* Prime 1 is 2, Prime 2 is 3 */
+        ret
+
+found_prime:
+        addi    a0, a0, -1
+        bnez    a0, outer
+        move    a0, t0
+        ret
+
+outer:
+        add     t0, t0, t1              /* prime candidate */
+        sub     t1, a1, t1              /* prime step */
+        li      t3, 4                   /* factor step */
+        li      t2, 1                   /* factor candidate will be 5 */
+
+inner:
+        add     t2, t2, t3              /* factor candidate */
+        sub     t3, a1, t3              /* factor step */
+        mul     t4, t2, t2
+        bgt     t4, t0, found_prime
+        rem     t5, t0, t2
+        bnez    t5, inner
+        j       outer

exercises/practice/nth-prime/.meta/tests.toml

@@ -0,0 +1,26 @@
+# This is an auto-generated file.
+#
+# Regenerating this file via `configlet sync` will:
+# - Recreate every `description` key/value pair
+# - Recreate every `reimplements` key/value pair, where they exist in problem-specifications
+# - Remove any `include = true` key/value pair (an omitted `include` key implies inclusion)
+# - Preserve any other key/value pair
+#
+# As user-added comments (using the # character) will be removed when this file
+# is regenerated, comments can be added via a `comment` key.
+
+[75c65189-8aef-471a-81de-0a90c728160c]
+description = "first prime"
+
+[2c38804c-295f-4701-b728-56dea34fd1a0]
+description = "second prime"
+
+[56692534-781e-4e8c-b1f9-3e82c1640259]
+description = "sixth prime"
+
+[fce1e979-0edb-412d-93aa-2c744e8f50ff]
+description = "big prime"
+
+[bd0a9eae-6df7-485b-a144-80e13c7d55b2]
+description = "there is no zeroth prime"
+include = false

exercises/practice/nth-prime/Makefile

@@ -0,0 +1,36 @@
+AS = /usr/bin/riscv64-linux-gnu-as
+CC = /usr/bin/riscv64-linux-gnu-gcc
+
+CFLAGS = -g -Wall -Wextra -pedantic -Werror
+LDFLAGS =
+
+ALL_LDFLAGS = -pie -Wl,--fatal-warnings
+
+ALL_CFLAGS = -std=c99 -fPIE $(CFLAGS)
+ALL_LDFLAGS += $(LDFLAGS)
+
+C_OBJS = $(patsubst %.c,%.o,$(wildcard *.c))
+AS_OBJS = $(patsubst %.s,%.o,$(wildcard *.s))
+ALL_OBJS = $(filter-out example.o,$(C_OBJS) $(AS_OBJS) vendor/unity.o)
+
+CC_CMD = $(CC) $(ALL_CFLAGS) -c -o $@ $<
+
+all: tests
+	qemu-riscv64 -L /usr/riscv64-linux-gnu ./$<
+
+tests: $(ALL_OBJS)
+	@$(CC) $(ALL_CFLAGS) $(ALL_LDFLAGS) -o $@ $(ALL_OBJS)
+
+%.o: %.s
+	@$(AS) -o $@ $<
+
+%.o: %.c
+	@$(CC_CMD)
+
+vendor/unity.o: vendor/unity.c vendor/unity.h vendor/unity_internals.h
+	@$(CC_CMD)
+
+clean:
+	@rm -f *.o vendor/*.o tests
+
+.PHONY: all clean

exercises/practice/nth-prime/nth_prime.s

@@ -0,0 +1,5 @@
+.text
+.globl prime
+
+prime:
+        ret

exercises/practice/nth-prime/nth_prime_test.c

@@ -0,0 +1,49 @@
+#include "vendor/unity.h"
+
+extern unsigned prime(unsigned number);
+
+void setUp(void) {
+}
+
+void tearDown(void) {
+}
+
+void test_first_prime(void) {
+    TEST_ASSERT_EQUAL_UINT(2, prime(1));
+}
+
+void test_second_prime(void) {
+    TEST_IGNORE();
+    TEST_ASSERT_EQUAL_UINT(3, prime(2));
+}
+
+void test_sixth_prime(void) {
+    TEST_IGNORE();
+    TEST_ASSERT_EQUAL_UINT(13, prime(6));
+}
+
+void test_big_prime(void) {
+    TEST_IGNORE();
+    TEST_ASSERT_EQUAL_UINT(104743, prime(10001));
+}
+
+void test_seventh_prime(void) {
+    TEST_IGNORE();
+    TEST_ASSERT_EQUAL_UINT(17, prime(7));
+}
+
+void test_very_big_prime(void) {
+    TEST_IGNORE();
+    TEST_ASSERT_EQUAL_UINT(821647, prime(65537));
+}
+
+int main(void) {
+    UNITY_BEGIN();
+    RUN_TEST(test_first_prime);
+    RUN_TEST(test_second_prime);
+    RUN_TEST(test_sixth_prime);
+    RUN_TEST(test_big_prime);
+    RUN_TEST(test_seventh_prime);
+    RUN_TEST(test_very_big_prime);
+    return UNITY_END();
+}