190 lines
6.8 KiB
Python
190 lines
6.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
import numpy as np
|
|
from random import shuffle, choice
|
|
from copy import deepcopy
|
|
|
|
# 0 - 8: means 0-8 mines, not opened
|
|
# opened block = the value of not opened block + 10
|
|
# 9 is mine, not opened
|
|
# 19 is flagged mine
|
|
# 20 is stepped mine
|
|
|
|
# default:
|
|
# HEIGHT = 8
|
|
# WIDTH = 8
|
|
# MINES = 9
|
|
|
|
IS_MINE = 9
|
|
DEAD = 20
|
|
|
|
def check_params(height, width, mines):
|
|
if height <= 0 or width <= 0:
|
|
return (False, "地图太小!")
|
|
elif mines > height * width:
|
|
return (False, "放不下这么多雷嘛!")
|
|
elif mines == height * width:
|
|
return (False, "一点就爆炸，有意思吗？")
|
|
elif mines < 0:
|
|
return (False, "暂时还不会放负数颗雷呢。")
|
|
elif mines <= height * width:
|
|
return (True, "")
|
|
def get_row_col(width, index):
|
|
row = index // width
|
|
col = index - width * row
|
|
return (row, col)
|
|
def get_index(width, row_col):
|
|
(row, col) = row_col
|
|
index = width * row + col
|
|
return index
|
|
class Board():
|
|
def __init__(self, height, width, mines):
|
|
self.height = height
|
|
self.width = width
|
|
self.mines = mines
|
|
self.map = None
|
|
self.mmap = None
|
|
self.moves = list()
|
|
self.state = 0 # 0:not playing, 1:playing, 2:win, 3:dead
|
|
# statistics
|
|
self.__op = 0
|
|
self.__is = 0
|
|
self.__3bv = 0
|
|
def __gen_map(self, first_move):
|
|
height = self.height
|
|
width = self.width
|
|
mines = self.mines
|
|
if mines >= height * width:
|
|
return
|
|
elif mines < 0:
|
|
return
|
|
# first_move should't be a mine, and if possible, it should be an open.
|
|
self.map = np.zeros((height, width), dtype=np.int8)
|
|
map_1d = [IS_MINE] * mines
|
|
zero_blocks = list()
|
|
fm_index = get_index(width, first_move)
|
|
zero_blocks.append(fm_index)
|
|
fm_nbrs = [rc for rc in self.__iter_neighbour(*first_move)]
|
|
if height * width - mines - 1 >= len(fm_nbrs):
|
|
fm_nbrs_index = [get_index(width, fm_nbr) for fm_nbr in fm_nbrs]
|
|
zero_blocks += fm_nbrs_index
|
|
map_1d += [0] * (height * width - mines - 1 - len(fm_nbrs))
|
|
else:
|
|
map_1d += [0] * (height * width - mines - 1)
|
|
shuffle(map_1d)
|
|
for mindex in sorted(zero_blocks):
|
|
map_1d.insert(mindex, 0)
|
|
for mindex in range(len(map_1d)):
|
|
if map_1d[mindex] == IS_MINE:
|
|
(row, col) = get_row_col(width, mindex)
|
|
self.map[row][col] = IS_MINE
|
|
for row in range(height):
|
|
for col in range(width):
|
|
if self.map[row][col] != IS_MINE:
|
|
mine_count = 0
|
|
for nbr_value in self.__iter_neighbour(row, col, return_rc=False):
|
|
if nbr_value == IS_MINE:
|
|
mine_count += 1
|
|
self.map[row][col] = mine_count
|
|
self.mmap = deepcopy(self.map)
|
|
def __iter_neighbour(self, row, col, return_rc=True):
|
|
height = self.height
|
|
width = self.width
|
|
for i in [a - 1 for a in range(3)]:
|
|
for j in [b - 1 for b in range(3)]:
|
|
if (i != 0 or j != 0) and row + i >= 0 and row + i <= height - 1 and \
|
|
col + j >= 0 and col + j <= width - 1:
|
|
if return_rc:
|
|
yield (row + i, col + j)
|
|
else:
|
|
yield self.map[row + i][col + j]
|
|
def __do_i_win(self):
|
|
unopened = 0
|
|
mines_opened = 0
|
|
for x in np.nditer(self.map):
|
|
if x <= 8:
|
|
unopened += 1
|
|
elif x in (19, DEAD):
|
|
mines_opened += 1
|
|
if mines_opened == self.mines:
|
|
return True
|
|
elif unopened == 0:
|
|
return True
|
|
else:
|
|
return False
|
|
def __open(self, row, col, automatic=False):
|
|
if self.state != 1:
|
|
return
|
|
if not automatic and self.map[row][col] == 9:
|
|
self.map[row][col] = DEAD
|
|
self.state = 3
|
|
return
|
|
elif self.map[row][col] == 0:
|
|
self.map[row][col] += 10 # open this block
|
|
# open other blocks
|
|
for nbr in self.__iter_neighbour(row, col):
|
|
self.__open(nbr[0], nbr[1], automatic=True)
|
|
elif self.map[row][col] >= 10:
|
|
# already opened
|
|
if automatic:
|
|
return
|
|
neighbour_mine_opened = 0
|
|
neighbour_unopened = 0
|
|
for neighbour in self.__iter_neighbour(row, col, return_rc=False):
|
|
if neighbour in (19, DEAD):
|
|
neighbour_mine_opened += 1
|
|
if neighbour <= 9:
|
|
neighbour_unopened += 1
|
|
if (neighbour_mine_opened == self.map[row][col] - 10) or \
|
|
(neighbour_unopened == self.map[row][col] - 10 - neighbour_mine_opened):
|
|
for nbr in self.__iter_neighbour(row, col):
|
|
self.__open(nbr[0], nbr[1], automatic=True)
|
|
else:
|
|
self.map[row][col] += 10
|
|
if self.__do_i_win():
|
|
self.state = 2
|
|
|
|
def move(self, row_col):
|
|
if self.state == 0:
|
|
self.__gen_map(row_col)
|
|
self.state = 1
|
|
(row, col) = row_col
|
|
self.__open(row, col)
|
|
|
|
def gen_statistics(self):
|
|
if self.__3bv != 0:
|
|
return (self.__op, self.__is, self.__3bv)
|
|
self.__visited = np.zeros((self.height, self.width), dtype=np.int8)
|
|
def scan_open(row, col):
|
|
self.__visited[row][col] = 1
|
|
for nbr_rc in self.__iter_neighbour(row, col):
|
|
(nrow, ncol) = nbr_rc
|
|
if self.__visited[nrow][ncol] == 0:
|
|
nbr = self.mmap[nrow][ncol]
|
|
if nbr == 0:
|
|
scan_open(nrow, ncol)
|
|
elif nbr <= 8:
|
|
self.__visited[nrow][ncol] = 1
|
|
def scan_island(row, col):
|
|
self.__3bv += 1
|
|
self.__visited[row, col] = 1
|
|
for nbr_rc in self.__iter_neighbour(row, col):
|
|
(nrow, ncol) = nbr_rc
|
|
if self.__visited[nrow][ncol] == 0:
|
|
nbr = self.mmap[nrow][ncol]
|
|
if nbr >= 1 and nbr <= 8:
|
|
scan_island(nrow, ncol)
|
|
|
|
for row in range(self.height):
|
|
for col in range(self.width):
|
|
if self.__visited[row][col] == 0 and self.mmap[row][col] == 0:
|
|
self.__op += 1
|
|
self.__3bv += 1
|
|
scan_open(row, col)
|
|
for row in range(self.height):
|
|
for col in range(self.width):
|
|
cell = self.mmap[row][col]
|
|
if self.__visited[row][col] == 0 and cell >= 1 and cell <= 8:
|
|
self.__is += 1
|
|
scan_island(row, col)
|
|
return (self.__op, self.__is, self.__3bv)
|