added CMakeLists.txt and simplified code

CMakeLists.txt

@@ -0,0 +1,5 @@
+cmake_minimum_required(VERSION 3.10)
+
+project(garland VERSION 1.0 LANGUAGES C)
+
+add_executable(garland main.c)

main.c

@@ -2,14 +2,16 @@
 #include <string.h>
 #include <stdio.h>
 
-#define YELLOW "\033[1;33m"
-#define RED "\033[1;31m"
-#define BLUE "\033[1;34m"
-#define GREEN "\033[1;32m"
+#define YELLOW "\033[1;33m#"
+#define RED "\033[1;31m#"
+#define BLUE "\033[1;34m#"
+#define GREEN "\033[1;32m#"
 #define RESET "\033[0m"
 
+static char* lightbulbs[] = {RED, YELLOW, BLUE, GREEN};
+
 typedef struct Garland {
-	char** lightbulbs;
+	char* lightbulbs;
 	unsigned int length;
 } Garland;
 
@@ -26,34 +28,14 @@ Garland generate_garland(unsigned int length) {
 	g.lightbulbs = malloc(length * sizeof(char*));
 	g.length = length;
 	for (unsigned int i = 0; i < length; i ++) {
-		char* color = (char*) malloc(9);
-		switch (rand() % 4) {
-			case 0:
-				strncpy(color, RED, 7);
-				break;
-			case 1:
-				strncpy(color, YELLOW, 7);
-				break;
-			case 2:
-				strncpy(color, GREEN, 7);
-				break;
-			case 3:
-				strncpy(color, BLUE, 7);
-				break;
-		}
-		strncpy(color+7, "#\0", 2);
-		
-		g.lightbulbs[i] = malloc(9);
-		strncpy(g.lightbulbs[i], color, 9);
-		free(color);
-		color = NULL;
+		g.lightbulbs[i] = rand() % 4;
 	}
 	return g;
 }
 
 void print_garland(Garland* g) {
 	for (unsigned int i = 0; i < g->length; i ++) {
-		printf("%s", g->lightbulbs[i]);
+		printf("%s", lightbulbs[g->lightbulbs[i]]);
 	}
 	printf(RESET);
 	printf("\n");