I hate SIGSEGV

day9/day.cpp

@@ -5,17 +5,17 @@
 #include <iostream>
 #include <vector>
 
-void printDiskMap(std::vector<char> &diskmap) {
+void printDiskMap(std::vector<int> &diskmap) {
   std::string numC = BOLDGREEN;
   for (int i = 0; i < diskmap.size(); i++) {
-    if (diskmap[i] == '.') {
+    if (diskmap[i] == -1) {
       numC = BOLDYELLOW;
       std::clog << RED "." << RESET;
     } else {
       std::clog << numC << diskmap[i] << RESET;
     }
   }
-  std::cout << RESET << std::endl;
+  std::clog << RESET << std::endl;
 }
 int main() {
   std::ifstream inputfile("input");
@@ -26,7 +26,7 @@ int main() {
     return ENOENT;
   }
 
-  std::vector<char> diskmap;
+  std::vector<int> diskmap;
 
   // create the disk map
   {
@@ -38,9 +38,9 @@ int main() {
 
       for (int j = 0; j < lineSplited[i] - '0'; j++) {
         if (i % 2 == 0) {
-          diskmap.push_back(id + '0');
+          diskmap.push_back(id);
         } else {
-          diskmap.push_back('.');
+          diskmap.push_back(-1);
         }
       }
       if (i % 2 == 0) {
@@ -48,24 +48,20 @@ int main() {
       }
     }
   }
-  printDiskMap(diskmap);
+  // printDiskMap(diskmap);
   {
     int firstDot;
-    while (std::find(diskmap.begin(), diskmap.end(), '.') != diskmap.end()) {
+    while (std::find(diskmap.begin(), diskmap.end(), -1) != diskmap.end()) {
       firstDot =
-          std::find(diskmap.begin(), diskmap.end(), '.') - diskmap.begin();
+          std::find(diskmap.begin(), diskmap.end(), -1) - diskmap.begin();
       diskmap[firstDot] = diskmap.back();
       diskmap.pop_back();
-      printDiskMap(diskmap);
+      // printDiskMap(diskmap);
     }
   }
-  std::vector<int> diskmapInt;
 
-  std::transform(diskmap.begin(), diskmap.end(), std::back_inserter(diskmapInt),
-                 [](const char &s) { return s - '0'; });
-
-  for (int i = 0; i < diskmapInt.size(); i++) {
-    res += i * diskmapInt[i];
+  for (int i = 0; i < diskmap.size(); i++) {
+    res += i * diskmap[i];
   }
   std::cout << CYAN "\nfilesystem checksum is: " << BOLDYELLOW << res << RESET
             << std::endl;

day9/second.cpp

@@ -0,0 +1,95 @@
+#include "../include/color.h"
+#include "../include/useful_funcs.h"
+#include <cstdlib>
+#include <fstream>
+#include <iostream>
+#include <vector>
+
+void printDiskMap(std::vector<int> &diskmap) {
+  std::string numC = BOLDGREEN;
+  for (int i = 0; i < diskmap.size(); i++) {
+    if (diskmap[i] == -1) {
+      numC = BOLDYELLOW;
+      std::clog << RED "." << RESET;
+    } else {
+      std::clog << numC << diskmap[i] << RESET;
+    }
+  }
+  std::clog << RESET << std::endl;
+}
+int main() {
+  std::ifstream inputfile("example");
+
+  unsigned long long res = 0;
+  if (!inputfile.is_open()) {
+    std::cerr << "Could not open the file" << std::endl;
+    return ENOENT;
+  }
+
+  std::vector<int> diskmap;
+
+  // create the disk map
+  {
+    std::string line;
+    inputfile >> line;
+    int id = 0;
+    std::vector<char> lineSplited = std::vector<char>(line.begin(), line.end());
+    for (int i = 0; i < lineSplited.size(); i++) {
+
+      for (int j = 0; j < lineSplited[i] - '0'; j++) {
+        if (i % 2 == 0) {
+          diskmap.push_back(id);
+        } else {
+          diskmap.push_back(-1);
+        }
+      }
+      if (i % 2 == 0) {
+        id++;
+      }
+    }
+  }
+  printDiskMap(diskmap);
+
+  {
+    int dotFound;
+    auto offset = diskmap.begin();
+    int block_count;
+
+    while (std::find(diskmap.begin(), diskmap.end(), -1) != diskmap.end()) {
+      // Removes empty blocks at the end
+      if (diskmap.back() == -1) {
+        diskmap.pop_back();
+        continue;
+      }
+
+      // counts how many blocks are in file
+      block_count = std::count(diskmap.end(), diskmap.begin(), diskmap.back());
+
+      while (true) {
+        dotFound = std::find(offset, diskmap.end(), -1) - diskmap.begin();
+
+        if (std::count(diskmap.begin() + dotFound,
+                       diskmap.begin() + dotFound + block_count,
+                       -1) == block_count) {
+
+          std::fill(diskmap.begin() + dotFound,
+                    diskmap.begin() + dotFound + block_count, diskmap.back());
+          for (int i = 0; i < block_count; i++) {
+            diskmap.pop_back();
+          }
+          break;
+        } else {
+          offset = diskmap.begin() + dotFound + block_count;
+        }
+      }
+
+      printDiskMap(diskmap);
+    }
+  }
+
+  for (int i = 0; i < diskmap.size(); i++) {
+    res += i * diskmap[i];
+  }
+  std::cout << CYAN "\nfilesystem checksum is: " << BOLDYELLOW << res << RESET
+            << std::endl;
+}