Support benchmarking

2 files changed, 4 insertions, 4 deletions

diff --git a/3/part1.c b/3/part1.c
index b6d209a..0373328 100644
--- a/3/part1.c
+++ b/3/part1.c
@@ -31,7 +31,7 @@ int count_trees(const char *filename)
     return hit;
 }
 
-int main()
+int main(int argc, char *argv[])
 {
-    printf("%i", count_trees("input"));
+    printf("%i", count_trees(argv[1]));
 }
diff --git a/3/part2.c b/3/part2.c
index 9214c25..7362c12 100644
--- a/3/part2.c
+++ b/3/part2.c
@@ -33,8 +33,8 @@ int count_trees(int inc_right, int inc_down, const char *filename)
     return hit;
 }
 
-int main()
+int main(int argc, char *argv[])
 {
-    long mul = (long)count_trees(1, 1, "input") * count_trees(3, 1, "input") * count_trees(5, 1, "input") * count_trees(7, 1, "input") * count_trees(1, 2, "input");
+    long mul = (long)count_trees(1, 1, argv[1]) * count_trees(3, 1, argv[1]) * count_trees(5, 1, argv[1]) * count_trees(7, 1, argv[1]) * count_trees(1, 2, argv[1]);
     printf("%li\n", mul);
 }