day06

1 files changed, 47 insertions, 0 deletions

diff --git a/day06/day06.cpp b/day06/day06.cpp
new file mode 100644
index 0000000..38d602b
--- /dev/null
+++ b/day06/day06.cpp
@@ -0,0 +1,47 @@
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <vector>
+
+#include "../utils.h"
+
+long sum (std::vector<long> vec) {
+	long s = 0; for (long v : vec) s += v; return s;
+}
+
+int main (int argc, char *argv[]) {
+	std::string raw;
+	std::getline(std::ifstream(argv[1]), raw, '\0');
+	std::vector<std::string> strfish;
+	split(strfish, raw, ",");
+	
+	std::vector<long> fish;
+	for (int i=0; i<9; i++) { fish.push_back(0); }
+	for (std::string str : strfish) fish[std::stoi(str)] += 1;
+
+	int days = 80;
+	if (argc > 2) days = std::stoi(std::string(argv[2]));
+	for (int i=0; i<days; i++) {
+		long born = fish[0];
+		fish[0] = 0;
+		for (int j=1; j<9; j++) {
+			fish[j-1] += fish[j];
+			fish[j] = 0;
+		}
+		fish[6] += born;
+		fish[8] = born;
+	
+		if (argc > 3 && std::string(argv[3]) == "-w") {	
+			std::cout << "day " << i+1 << "/" << days 
+				<< " current fish: " << sum(fish) << std::endl;
+		}
+		if (argc > 3 && std::string(argv[3]) == "-v") {	
+			std::cout << "day " << i+1 << ": ";
+			for (auto f : fish) std::cout << f << ",";
+			std::cout << std::endl;
+		}
+	}
+
+	std::cout << "fish: " << sum(fish) << std::endl;
+	return 0;
+}