69 lines
1.6 KiB
C++
69 lines
1.6 KiB
C++
#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("input");
|
|
|
|
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 firstDot;
|
|
while (std::find(diskmap.begin(), diskmap.end(), -1) != diskmap.end()) {
|
|
firstDot =
|
|
std::find(diskmap.begin(), diskmap.end(), -1) - diskmap.begin();
|
|
diskmap[firstDot] = diskmap.back();
|
|
diskmap.pop_back();
|
|
// 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;
|
|
}
|