eryn/sudoku
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add peers and index_peers for a given (x,y)

Browse source
This commit is contained in:
Eryn Wells 2017-10-10 09:55:51 -07:00
parent 19a9bc9d37
commit 00225aca44
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
sudoku.py
View file

@ -60,6 +60,16 @@ class Sudoku:
dim = self.dimension dim = self.dimension
return (self._square(x, y, dim) for y in range(dim) for x in range(dim)) return (self._square(x, y, dim) for y in range(dim) for x in range(dim))
def peers(self, x, y):
return {self._board[i] for i in self.index_peers(x, y) if self._board[i] != 0}
def index_peers(self, x, y):
dim = self.dimension
sz = self.size
sz2 = sz ** 2
sq_x, sq_y = int(x / dim), int(y / dim)
return set(self._row(y, sz)) | set(self._column(x, sz, sz2)) | set(self._square(sq_x, sq_y, dim))
# TODO: Break the above into helper methods that produce a single thing given an index. # TODO: Break the above into helper methods that produce a single thing given an index.
def _row(self, r, size): def _row(self, r, size):
return range(r * size, r * size + size) return range(r * size, r * size + size)