109 lines
2.1 KiB
C++

#include "../include/useful_funcs.h"
#include <fstream>
#include <iostream>
#include <ostream>
#include <vector>
bool WillHeGetStuck(int posX, int posY, int direction, int locX, int locY,
std::vector<std::vector<char>> map) {
char op_x;
char op_y;
map[locX][locY] = '#';
int conter = 0;
while (true) {
map[posX][posY] = 'X';
conter++;
if (conter > 10000) {
return true;
}
switch (direction) {
case 0:
op_x = '-';
op_y = 'n';
break;
case 1:
op_x = 'n';
op_y = '+';
break;
case 2:
op_x = '+';
op_y = 'n';
break;
case 3:
op_x = 'n';
op_y = '-';
break;
default:
std::cerr << "How did we get here?" << std::endl;
exit(255);
break;
}
if (!AreWeInBounds(condop(posX, 1, op_x), condop(posY, 1, op_y), map)) {
break;
}
switch (map[condop(posX, 1, op_x)][condop(posY, 1, op_y)]) {
case '#':
direction++;
if (direction > 3) {
direction = 0;
}
break;
default:
posX = condop(posX, 1, op_x);
posY = condop(posY, 1, op_y);
}
}
return false;
}
int main() {
std::ifstream inputfile("input");
std::string input;
int res = 0;
if (!inputfile.is_open()) {
std::cerr << "Could not open the file" << std::endl;
return ENOENT;
}
std::string line;
std::vector<std::vector<char>> map;
while (std::getline(inputfile, line)) {
map.push_back(std::vector<char>(line.begin(), line.end()));
}
int posX;
int posY;
int direction;
/*
0: up
1: right
2: down
3: left
*/
for (int i = 0; i < map.size(); i++) {
for (int j = 0; j < map[i].size(); j++) {
if (map[i][j] == '^') {
posX = i;
posY = j;
direction = 0;
}
}
}
for (int i = 0; i < map.size(); i++) {
for (int j = 0; j < map[i].size(); j++) {
if (map[i][j] == '^' || map[i][j] == '#') {
continue;
} else {
if (WillHeGetStuck(posX, posY, direction, i, j, map)) {
res++;
}
}
}
}
std::cout << "\nRes: " << res << std::endl;
}