diff options
Diffstat (limited to '2021/15/solve.c')
| -rw-r--r-- | 2021/15/solve.c | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/2021/15/solve.c b/2021/15/solve.c index c0c16cb..77619cb 100644 --- a/2021/15/solve.c +++ b/2021/15/solve.c @@ -89,7 +89,7 @@ static int part_one(FILE *fp) int visited[SIZE][SIZE] = { 0 }; - struct queue *queue = create(500); + struct queue *queue = create(1000); push(queue, 0, 0, 0); while (1) { @@ -109,6 +109,10 @@ static int part_one(FILE *fp) push(queue, x + 1, y, risk + GET(x + 1, y)); if (y + 1 < SIZE && !visited[x][y + 1]) push(queue, x, y + 1, risk + GET(x, y + 1)); + if (x - 1 >= 0 && !visited[x - 1][y]) + push(queue, x - 1, y, risk + GET(x - 1, y)); + if (y - 1 >= 0 && !visited[x][y - 1]) + push(queue, x, y - 1, risk + GET(x, y - 1)); } free(data); @@ -131,7 +135,7 @@ static int part_two(FILE *fp) int visited[SIZE2][SIZE2] = { 0 }; - struct queue *queue = create(1500); + struct queue *queue = create(10000); push(queue, 0, 0, 0); while (1) { @@ -143,16 +147,18 @@ static int part_two(FILE *fp) if (visited[x][y]) continue; - if (x == SIZE2 - 1 && y == SIZE2 - 1) { - risk -= GET2(x, y); // Hmmm + if (x == SIZE2 - 1 && y == SIZE2 - 1) break; - } visited[x][y] = 1; if (x + 1 < SIZE2 && !visited[x + 1][y]) push(queue, x + 1, y, risk + GET2(x + 1, y)); if (y + 1 < SIZE2 && !visited[x][y + 1]) push(queue, x, y + 1, risk + GET2(x, y + 1)); + if (x - 1 >= 0 && !visited[x - 1][y]) + push(queue, x - 1, y, risk + GET2(x - 1, y)); + if (y - 1 >= 0 && !visited[x][y - 1]) + push(queue, x, y - 1, risk + GET2(x, y - 1)); } free(data); |