mia
/
aoc2022
1
0
Fork 0
Code Releases Activity

05: C: ~6x faster implementation

This commit is contained in:
Mia Herkt 2022-12-05 23:43:58 +01:00
parent f1edb14ad5
commit af88bca186
Signed by: mia
GPG Key ID: 72E154B8622EC191
1 changed files with 51 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
05/solution.c
View File

@ -3,70 +3,79 @@
#include <string.h> #include <string.h>
#include <ctype.h> #include <ctype.h>
typedef struct crate { typedef struct crates {
char c; char *c;
struct crate *nA; size_t len, alloc;
struct crate *nB; } crates;
} crate;
#define alloc_stacks(C, N) \ #define alloc_stacks(C) \
if (C) C = realloc(C, sizeof(crate) * N); \ if (C) C = realloc(C, sizeof(crates) * A); \
else C = malloc(sizeof(crate) * N); else C = malloc(sizeof(crates) * A); \
for (i = N; i < A; i++) C[i].c = 0;
#define alloc_crates(C, N) \
if (C.c) { \
C.len += N; \
if (C.len > C.alloc) \
C.c = realloc(C.c, C.alloc = C.len); \
} else \
C.c = malloc(C.len = C.alloc = N);
#define crate_pos(C, N) (C.c + C.len + N)
int main(void) { int main(void) {
char l[BUFSIZ], *e, *m = 0; char l[BUFSIZ], *e, *m = 0, ic, *tc, *tf;
crate **A = 0, **B = 0, *c, *p; crates *stacksA = 0, *stacksB = 0;
int i, n, N = 0, f, t; int i, n, N = 0, A, f, t;
while (fgets(l, BUFSIZ, stdin)) { while (fgets(l, BUFSIZ, stdin)) {
if ((e = strrchr(l, ']'))) { if ((e = strrchr(l, ']'))) {
if (e > m) { if (e > m) {
m = e; m = e;
N = (m - l) / 4 + 1; A = (m - l) / 4 + 1;
alloc_stacks(A, N); alloc_stacks(stacksA)
alloc_stacks(B, N); alloc_stacks(stacksB)
N = A;
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
c = calloc(1, sizeof(crate)); if ((ic = l[i*4+1]) != ' ') {
c->c = l[i*4+1]; alloc_crates(stacksA[i], 1)
alloc_crates(stacksB[i], 1)
if (!A[i] || A[i]->c == ' ') { memmove(stacksA[i].c + 1, stacksA[i].c, stacksA[i].len - 1);
A[i] = c; memmove(stacksB[i].c + 1, stacksB[i].c, stacksB[i].len - 1);
B[i] = c; stacksA[i].c[0] = ic;
} else { stacksB[i].c[0] = ic;
for (p = A[i]; p->nA; p = p->nA);
p->nA = c;
for (p = B[i]; p->nB; p = p->nB);
p->nB = c;
} }
} }
} }
if (sscanf(l, "move %u from %u to %u", &n, &f, &t) == 3) { if (l[0] == 'm') {
f--; n = strtol(l + 5, &e, 10);
t--; f = strtol(e + 6, &e, 10) - 1;
for (i = 0; i < n; i++) { t = atoi(e + 4) - 1;
c = A[f];
A[f] = c->nA;
c->nA = A[t];
A[t] = c;
}
c = B[f]; alloc_crates(stacksA[t], n)
for (i = 1; i < n; i++) c = c->nB; tf = crate_pos(stacksA[f], -1);
p = c->nB; tc = crate_pos(stacksA[t], -n);
c->nB = B[t]; for (i = 0; i < n; i++)
B[t] = B[f]; tc[i] = tf[-i];
B[f] = p; stacksA[f].len -= n;
alloc_crates(stacksB[t], n)
tf = crate_pos(stacksB[f], -n);
tc = crate_pos(stacksB[t], -n);
for (i = 0; i < n; i++)
tc[i] = tf[i];
stacksB[f].len -= n;
} }
} }
printf("Silver: "); printf("Silver: ");
for (i = 0; i < N; i++) for (i = 0; i < N; i++)
if (A[i]) printf("%c", A[i]->c); printf("%c", *crate_pos(stacksA[i], -1));
printf("\nGold: "); printf("\nGold: ");
for (i = 0; i < N; i++) for (i = 0; i < N; i++)
if (B[i]) printf("%c", B[i]->c); printf("%c", *crate_pos(stacksB[i], -1));
printf("\n"); printf("\n");
} }