aboutsummaryrefslogtreecommitdiff
path: root/2023/17
diff options
context:
space:
mode:
authorSebastiano Tronto <sebastiano@tronto.net>2023-12-19 15:54:12 +0100
committerSebastiano Tronto <sebastiano@tronto.net>2023-12-19 15:54:12 +0100
commit1f75d7c9ebec207c8193754792076dc66eba9713 (patch)
tree48556a419f050153513cd5a3b8a4a6e125ba2c45 /2023/17
parent83adb5e4215a25d17cfc11b4dbf10cf20c418c20 (diff)
downloadaoc-1f75d7c9ebec207c8193754792076dc66eba9713.tar.gz
aoc-1f75d7c9ebec207c8193754792076dc66eba9713.zip
Added solutions for 15, 16, 17, 18 and 19
Diffstat (limited to '2023/17')
-rw-r--r--2023/17/17a.c73
-rw-r--r--2023/17/17b.c74
2 files changed, 147 insertions, 0 deletions
diff --git a/2023/17/17a.c b/2023/17/17a.c
new file mode 100644
index 0000000..0b02f45
--- /dev/null
+++ b/2023/17/17a.c
@@ -0,0 +1,73 @@
1#include <stdbool.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5
6#define M 150
7#define MAXE 1000000
8#define INF 999999999
9
10#define E 0
11#define N 1
12#define W 2
13#define S 3
14
15int go[4][2] = { [E] = {0, 1}, [N] = {-1, 0}, [W] = {0, -1}, [S] = {1, 0} };
16
17typedef struct { int i, j, k, s, d; } node_t;
18
19char map[M][M];
20int n, nq, d[M][M][4][4];
21node_t heap[MAXE];
22
23void swap(int i, int j) {
24 node_t aux = heap[i];
25 heap[i] = heap[j];
26 heap[j] = aux;
27}
28
29void push(node_t node) {
30 heap[nq] = node;
31 for (int k = nq++; k > 0 && heap[k].d < heap[(k-1)/2].d; k = (k-1)/2)
32 swap(k, (k-1)/2);
33}
34
35node_t pop(void) {
36 node_t ret = heap[0];
37 heap[0] = heap[--nq];
38 for (int k = 0, j = 0; 2*k+1 < nq; k = j) {
39 j = 2*k+1;
40 if (j+1 < nq && heap[j].d > heap[j+1].d) j++;
41 if (heap[k].d > heap[j].d) swap(k, j);
42 }
43 return ret;
44}
45
46int main() {
47 for (n = 0; fgets(map[n], M, stdin) != NULL; n++) ;
48
49 for (int i = 0; i < n; i++)
50 for (int j = 0; j < n; j++)
51 for (int k = 0; k < 4; k++)
52 for (int s = 0; s < 4; s++)
53 d[i][j][k][s] = INF;
54
55 push((node_t){.i = 0, .j = 0, .k = -1, .s = 0});
56 node_t v;
57 for (v = pop(); v.i != n-1 || v.j != n-1; v = pop()) {
58 for (int k = 0; k < 4; k++) {
59 if (k == v.k + 2 || k == v.k - 2) continue;
60 int ii = v.i+go[k][0];
61 int jj = v.j+go[k][1];
62 if (ii < 0 || jj < 0 || ii >= n || jj >= n) continue;
63 int ss = k == v.k ? v.s + 1 : 1;
64 int dd = v.d + map[ii][jj] - '0';
65 if (d[ii][jj][k][ss] <= dd || ss > 3) continue;
66 d[ii][jj][k][ss] = dd;
67 push((node_t){.i=ii, .j=jj, .k=k, .s=ss, .d=dd});
68 }
69 }
70
71 printf("%d\n", v.d);
72 return 0;
73}
diff --git a/2023/17/17b.c b/2023/17/17b.c
new file mode 100644
index 0000000..10fad81
--- /dev/null
+++ b/2023/17/17b.c
@@ -0,0 +1,74 @@
1#include <stdbool.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5
6#define M 150
7#define MAXE 1000000
8#define INF 999999999
9
10#define E 0
11#define N 1
12#define W 2
13#define S 3
14
15int go[4][2] = { [E] = {0, 1}, [N] = {-1, 0}, [W] = {0, -1}, [S] = {1, 0} };
16
17typedef struct { int i, j, k, s, d; } node_t;
18
19char map[M][M];
20int n, nq, d[M][M][4][11];
21node_t heap[MAXE];
22
23void swap(int i, int j) {
24 node_t aux = heap[i];
25 heap[i] = heap[j];
26 heap[j] = aux;
27}
28
29void push(node_t node) {
30 heap[nq] = node;
31 for (int k = nq++; k > 0 && heap[k].d < heap[(k-1)/2].d; k = (k-1)/2)
32 swap(k, (k-1)/2);
33}
34
35node_t pop(void) {
36 node_t ret = heap[0];
37 heap[0] = heap[--nq];
38 for (int k = 0, j = 0; 2*k+1 < nq; k = j) {
39 j = 2*k+1;
40 if (j+1 < nq && heap[j].d > heap[j+1].d) j++;
41 if (heap[k].d > heap[j].d) swap(k, j);
42 }
43 return ret;
44}
45
46int main() {
47 for (n = 0; fgets(map[n], M, stdin) != NULL; n++) ;
48
49 for (int i = 0; i < n; i++)
50 for (int j = 0; j < n; j++)
51 for (int k = 0; k < 4; k++)
52 for (int s = 0; s < 11; s++)
53 d[i][j][k][s] = INF;
54
55 push((node_t){.i = 0, .j = 0, .k = -1, .s = 0});
56 node_t v;
57 for (v = pop(); v.i != n-1 || v.j != n-1; v = pop()) {
58 for (int k = 0; k < 4; k++) {
59 if (k == v.k + 2 || k == v.k - 2) continue;
60 if (v.s < 4 && v.k != -1 && v.k != k) continue;
61 int ii = v.i+go[k][0];
62 int jj = v.j+go[k][1];
63 if (ii < 0 || jj < 0 || ii >= n || jj >= n) continue;
64 int ss = k == v.k ? v.s + 1 : 1;
65 int dd = v.d + map[ii][jj] - '0';
66 if (d[ii][jj][k][ss] <= dd || ss > 10) continue;
67 d[ii][jj][k][ss] = dd;
68 push((node_t){.i=ii, .j=jj, .k=k, .s=ss, .d=dd});
69 }
70 }
71
72 printf("%d\n", v.d);
73 return 0;
74}

Generated with cgit - Back to sebastiano.tronto.net