Refactor to extract update() function
This commit is contained in:
parent
001b52d714
commit
02a0dd9763
92
main.c
92
main.c
@ -17,6 +17,17 @@
|
|||||||
|
|
||||||
enum { UNKNOWN = 0xfe, KILLER = 0xfd };
|
enum { UNKNOWN = 0xfe, KILLER = 0xfd };
|
||||||
|
|
||||||
|
typedef enum {
|
||||||
|
FOUND_MINE,
|
||||||
|
FOUND_SAFE,
|
||||||
|
FOUND_NOTHING,
|
||||||
|
} update_res_t;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
int mines, unknown;
|
||||||
|
puzz_t field;
|
||||||
|
} state_t;
|
||||||
|
|
||||||
static void setadj(puzz_t field, int x, int y, uint8_t from, uint8_t to)
|
static void setadj(puzz_t field, int x, int y, uint8_t from, uint8_t to)
|
||||||
{
|
{
|
||||||
for (int yp = y - 1; yp < y + 2; ++yp) {
|
for (int yp = y - 1; yp < y + 2; ++yp) {
|
||||||
@ -46,53 +57,66 @@ static void getadj(puzz_t field, int *x, int *y, uint8_t val)
|
|||||||
assert(false);
|
assert(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
static status_t solve(void)
|
static update_res_t update(state_t *state, int *x_out, int *y_out)
|
||||||
{
|
{
|
||||||
puzz_t field;
|
state->mines = state->unknown = 0;
|
||||||
memset(field, UNKNOWN, sizeof(field));
|
for (int y = 0; y < HEIGHT; ++y) {
|
||||||
|
for (int x = 0; x < WIDTH; ++x) {
|
||||||
status_t status;
|
state->mines += state->field[x][y] == MINE ? 1 : 0;
|
||||||
int total_mines = 0, total_unknowns = WIDTH * HEIGHT;
|
state->unknown += state->field[x][y] == UNKNOWN ? 1 : 0;
|
||||||
do {
|
|
||||||
int x = rand() % WIDTH;
|
|
||||||
int y = rand() % HEIGHT;
|
|
||||||
|
|
||||||
probe:
|
|
||||||
if (field[x][y] != MINE && (status = probe(x, y, field)) == DEAD) {
|
|
||||||
field[x][y] = KILLER;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
update:
|
|
||||||
total_mines = total_unknowns = 0;
|
|
||||||
for (y = 0; y < HEIGHT; ++y) {
|
|
||||||
for (x = 0; x < WIDTH; ++x) {
|
|
||||||
total_mines += field[x][y] == MINE ? 1 : 0;
|
|
||||||
total_unknowns += field[x][y] == UNKNOWN ? 1 : 0;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for (y = 0; y < HEIGHT; ++y) {
|
for (int y = 0; y < HEIGHT; ++y) {
|
||||||
for (x = 0; x < WIDTH; ++x) {
|
for (int x = 0; x < WIDTH; ++x) {
|
||||||
if (field[x][y] == UNKNOWN || field[x][y] == MINE)
|
if (state->field[x][y] == UNKNOWN || state->field[x][y] == MINE)
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
const int mines = countadj(field, x, y, MINE);
|
const int mines = countadj(state->field, x, y, MINE);
|
||||||
const int unknowns = countadj(field, x, y, UNKNOWN);
|
const int unknowns = countadj(state->field, x, y, UNKNOWN);
|
||||||
if (unknowns == 0)
|
if (unknowns == 0)
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
if (mines + unknowns == field[x][y]) {
|
if (mines + unknowns == state->field[x][y]) {
|
||||||
setadj(field, x, y, UNKNOWN, MINE);
|
setadj(state->field, x, y, UNKNOWN, MINE);
|
||||||
goto update;
|
return FOUND_MINE;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (mines == field[x][y]) {
|
if (mines == state->field[x][y]) {
|
||||||
getadj(field, &x, &y, UNKNOWN);
|
getadj(state->field, &x, &y, UNKNOWN);
|
||||||
goto probe;
|
*x_out = x;
|
||||||
|
*y_out = y;
|
||||||
|
return FOUND_SAFE;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} while (total_mines < NMINES);
|
|
||||||
|
return FOUND_NOTHING;
|
||||||
|
}
|
||||||
|
|
||||||
|
static status_t solve(void)
|
||||||
|
{
|
||||||
|
state_t state = { .mines = 0, .unknown = WIDTH * HEIGHT };
|
||||||
|
memset(state.field, UNKNOWN, sizeof(puzz_t));
|
||||||
|
|
||||||
|
int x = rand() % WIDTH;
|
||||||
|
int y = rand() % HEIGHT;
|
||||||
|
|
||||||
|
status_t status;
|
||||||
|
do {
|
||||||
|
if (state.field[x][y] != MINE
|
||||||
|
&& (status = probe(x, y, state.field)) == DEAD)
|
||||||
|
break;
|
||||||
|
|
||||||
|
update_res_t update_res;
|
||||||
|
do
|
||||||
|
update_res = update(&state, &x, &y);
|
||||||
|
while (update_res == FOUND_MINE);
|
||||||
|
|
||||||
|
if (update_res == FOUND_NOTHING) {
|
||||||
|
x = rand() % WIDTH;
|
||||||
|
y = rand() % HEIGHT;
|
||||||
|
}
|
||||||
|
} while (state.mines < NMINES);
|
||||||
|
|
||||||
return status;
|
return status;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user