81 lines
1.3 KiB
C++
81 lines
1.3 KiB
C++
|
#include <BAN/Array.h>
|
||
|
#include <BAN/Vector.h>
|
||
|
|
||
|
#include <inttypes.h>
|
||
|
#include <stdio.h>
|
||
|
|
||
|
using i8 = int8_t;
|
||
|
using i16 = int16_t;
|
||
|
using i32 = int32_t;
|
||
|
using i64 = int64_t;
|
||
|
using isize = ssize_t;
|
||
|
|
||
|
using u8 = uint8_t;
|
||
|
using u16 = uint16_t;
|
||
|
using u32 = uint32_t;
|
||
|
using u64 = uint64_t;
|
||
|
using usize = size_t;
|
||
|
|
||
|
i64 part1(FILE* fp)
|
||
|
{
|
||
|
BAN::Vector<BAN::Array<u8, 5>> locks;
|
||
|
BAN::Vector<BAN::Array<u8, 5>> keys;
|
||
|
|
||
|
for (;;)
|
||
|
{
|
||
|
char buffer[6*7 + 1];
|
||
|
if (fread(buffer, 1, sizeof(buffer), fp) < sizeof(buffer) - 1)
|
||
|
break;
|
||
|
|
||
|
BAN::Array<u8, 5> obj;
|
||
|
for (usize j = 1; j <= 5; j++)
|
||
|
for (usize i = 0; i < 5; i++)
|
||
|
obj[i] += (buffer[6 * j + i] == '#');
|
||
|
|
||
|
MUST((buffer[0] == '#' ? locks : keys).push_back(obj));
|
||
|
}
|
||
|
|
||
|
u32 result = 0;
|
||
|
for (auto lock : locks)
|
||
|
{
|
||
|
for (auto key : keys)
|
||
|
{
|
||
|
bool valid = true;
|
||
|
for (usize i = 0; i < 5 && valid; i++)
|
||
|
valid = (key[i] + lock[i] <= 5);
|
||
|
result += valid;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
i64 part2(FILE* fp)
|
||
|
{
|
||
|
(void)fp;
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
int main(int argc, char** argv)
|
||
|
{
|
||
|
const char* file_path = "/usr/share/aoc2024/day25_input.txt";
|
||
|
|
||
|
if (argc >= 2)
|
||
|
file_path = argv[1];
|
||
|
|
||
|
FILE* fp = fopen(file_path, "r");
|
||
|
if (fp == nullptr)
|
||
|
{
|
||
|
perror("fopen");
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
printf("part1: %" PRId64 "\n", part1(fp));
|
||
|
|
||
|
fseek(fp, 0, SEEK_SET);
|
||
|
|
||
|
printf("part2: %" PRId64 "\n", part2(fp));
|
||
|
|
||
|
fclose(fp);
|
||
|
}
|