BlueRains

AdventOfCode C 2024 day 7

Dec 7th, 2024 (edited)
51
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 2.49 KB | None | 0 0
  1. #include <stdio.h>
  2. #include <stdbool.h>
  3. #define SIZE 850
  4. #define COL 20
  5.  
  6. int parse(FILE* file, long int* results, int items[][COL], int* sizes) {
  7.     int index = 0;
  8.     while (fscanf(file, "%ld: ", (results + index)) == 1) {
  9.         int val;
  10.         int offset = 0;
  11.         char c;
  12.         while (fscanf(file, "%d", &val) == 1) {
  13.             items[index][offset++] = val;
  14.             if ((c = fgetc(file)) != ' ') {
  15.                 // End of this string of values
  16.                 sizes[index] = offset;
  17.                 index++;
  18.                 offset = 0;
  19.                 break;
  20.             }
  21.         }
  22.     }
  23.     return index;
  24. }
  25.  
  26. bool evaluate(long int result, long int current, int* items, int size_remaining, bool part_one) {
  27.     if (size_remaining == 0) {
  28.         return result == current;
  29.     }
  30.     if (evaluate(result, current + *items, items + 1, size_remaining - 1)) {
  31.         return true;
  32.     }
  33.     if (evaluate(result, current * *items, items + 1, size_remaining - 1)) {
  34.         return true;
  35.     }
  36.     if (part_one) {
  37.         // Don't try ||
  38.         return false;
  39.     }
  40.     int val = *items;
  41.     // Get number of digits of val. shift current by that.
  42.     do {
  43.         val = val / 10;
  44.         current *= 10;
  45.     } while (val != 0);
  46.     current += *items;
  47.     bool concat = evaluate(result, current, items + 1, size_remaining - 1);
  48.     return concat;
  49. }
  50.  
  51. long int part_one(long int* results, int items[][COL], int* sizes, int size) {
  52.     long int total = 0;
  53.     for (size_t i = 0; i < size; i++) {
  54.         long int res = *(results + i);
  55.         int* values = items[i];
  56.         int size = *(sizes + i);
  57.         if (evaluate(res, *values, values + 1, size - 1, true)) {
  58.             total += res;
  59.         }
  60.     }
  61.     return total;
  62. }
  63. long int part_two(long int* results, int items[][COL], int* sizes, int size) {
  64.     long int total = 0;
  65.     for (size_t i = 0; i < size; i++) {
  66.         long int res = *(results + i);
  67.         int* values = items[i];
  68.         int size = *(sizes + i);
  69.         bool success = evaluate(res, *values, values + 1, size - 1, false);
  70.         if (success) {
  71.             total += res;
  72.         }
  73.     }
  74.     return total;
  75. }
  76.  
  77. int main() {
  78.     FILE* file = fopen("2024-7.txt", "r");
  79.     long int results[SIZE];
  80.     int sizes[SIZE];
  81.     int items[SIZE][COL];
  82.     int size = parse(file, results, items, sizes);
  83.     printf("Part one: %ld\n", part_one(results, items, sizes, size));
  84.     printf("Part two: %ld\n", part_two(results, items, sizes, size));
  85. }
Advertisement
Add Comment
Please, Sign In to add comment