Весела обнови решението на 29.11.2022 14:36 (преди над 1 година)
+class ChessException(Exception):
+ pass
+
+
+def has_adjacent_kings(rows):
По-добре е тази функция да е в класа, който я използва.
+ board = []
+
+ for index, row in enumerate(rows):
+ positions = [*(''.join([figure if not figure.isdigit() else " " * int(figure) for figure in row]))]
+ board.append(positions)
+
+ for rowIndex, row in enumerate(board):
Моля използвай snake_case за имена на променливи.
+ for colIndex, col in enumerate(row):
+ if col == " ":
+ continue
+ if col == "k":
+ if rowIndex != 0:
+ if board[rowIndex - 1][colIndex] == "K" or (
+ colIndex != 0 and board[rowIndex - 1][colIndex - 1] == "K") or (
+ colIndex != 7 and board[rowIndex - 1][colIndex + 1] == "K"):
+ return True
+ if rowIndex != 7:
+ if board[rowIndex + 1][colIndex] == "K" or (
+ colIndex != 0 and board[rowIndex + 1][colIndex - 1] == "K") or (
+ colIndex != 7 and board[rowIndex + 1][colIndex + 1] == "K"):
+ return True
+ if colIndex != 0:
+ if board[rowIndex][colIndex - 1] == "K":
+ return True
+ if colIndex != 7:
+ if board[rowIndex][colIndex + 1] == "K":
+ return True
+ return False
+
+
+def check_kings(rows):
+ is_white_king_available = False
+ is_black_king_available = False
+ white_king_count = 0
+ black_king_count = 0
+
+ for index, line in enumerate(rows):
+ if "K" in line:
+ is_white_king_available = True
+ white_king_count += 1
+ if "k" in line:
+ is_black_king_available = True
+ black_king_count += 1
+
+ if not is_black_king_available or not is_white_king_available or white_king_count > 1 or black_king_count > 1:
+ return False
+ return True
+
+
+def check_pawns(rows):
+ for index, line in enumerate(rows):
Можеш просто да обходиш само първия и последния ред, вместо да проверяваш индексите в последствие.
for line in rows[0] + rows[-1]
+ if ("P" in line or "p" in line) and (index == 0 or index == 7):
+ return True
+ return False
+
+
+class ChessPosition:
+ def __init__(self, fen):
+ rows = fen.split('/')
+ rows.reverse()
+
+ if has_adjacent_kings(rows):
+ raise ChessException("kings")
+ elif not check_kings(rows):
+ raise ChessException("kings")
+ elif check_pawns(rows):
+ raise ChessException("pawns")
+ self._fen = fen
+
+ def get_white_score(self):
+ white_figures_on_the_board = [figure for figure in self._fen if figure.isupper()]
+ white_score = ChessScore(white_figures_on_the_board)
+ return int(white_score)
Очаква се тези методи да връщаш ChessScore обект, а не интеджер.
+
+ def get_black_score(self):
+ black_figures_on_the_board = [figure for figure in self._fen if figure.islower()]
+ black_score = ChessScore(black_figures_on_the_board)
+ return int(black_score)
+
+ def white_is_winning(self):
Тук би било по-изчистено да използвам get_white_score и get_black_score, но реших да използвам предефинираните оператори
По-удачно би било, да. Освен това, добра идея е предварително да присвоиш в атрибут на инстанциите самите ChessScore обекти, за да не се налага да ги инициализираш при всяко използване на метод. Имайки ги като атрибути на инстанцията, просто ги сравняваш. Те самите вече са подготвени за сравнение, поради изискванията на задачата.
+ white_figures_on_the_board = [figure for figure in self._fen if figure.isupper()]
+ black_figures_on_the_board = [figure for figure in self._fen if figure.islower()]
+ white_score = ChessScore(white_figures_on_the_board)
+ black_score = ChessScore(black_figures_on_the_board)
+ return white_score > black_score
+
+ def black_is_winning(self):
+ white_figures_on_the_board = [figure for figure in self._fen if figure.isupper()]
+ black_figures_on_the_board = [figure for figure in self._fen if figure.islower()]
+ white_score = ChessScore(white_figures_on_the_board)
+ black_score = ChessScore(black_figures_on_the_board)
+ return white_score < black_score
+
+ def is_equal(self):
+ white_figures_on_the_board = [figure for figure in self._fen if figure.isupper()]
+ black_figures_on_the_board = [figure for figure in self._fen if figure.islower()]
+ white_score = ChessScore(white_figures_on_the_board)
+ black_score = ChessScore(black_figures_on_the_board)
+ return white_score == black_score
+
+ def __str__(self):
+ return self._fen
+
+ def __len__(self):
+ return len(''.join([figure for figure in self._fen if not figure.isdigit()]).replace('/', ''))
Можеш да използваш isalpha, което ще замени както not isdigit, така и replace('/', '').
+
+ def __getitem__(self, pos):
Бих дефинирал един речник предварително, базиран на самия fen, за да мога тук просто да потърся ключ, съвпадащ с pos. Реално този речник би ти свършил чудесна работа и за останалите методи.
+ letters_to_nums = {
+ "A": 0,
+ "B": 1,
+ "C": 2,
+ "D": 3,
+ "E": 4,
+ "F": 5,
+ "G": 6,
+ "H": 7
+ }
+
+ board = []
+ rows = self._fen.split('/')
+ rows.reverse()
+
+ for index, line in enumerate(rows):
+ positions = [*(''.join([figure if not figure.isdigit() else " " * int(figure) for figure in line]))]
+ board.append(positions)
+
+ return board[int(pos[1]) - 1][letters_to_nums[pos[0]]] if board[int(pos[1]) - 1][
+ letters_to_nums[pos[0]]] != " " else None
+
+
+class ChessScore:
+ def __init__(self, figures):
+ self._figures = figures
+
+ def sum_of_scores(self):
+ scores = {
+ "R": 5,
+ "N": 3,
+ "B": 3,
+ "Q": 9,
+ "K": 4,
+ "P": 1
+ }
+ sum_scores = 0
+
+ for figure in self._figures:
+ sum_scores += scores[figure.upper()]
+
+ return sum_scores
+
+ __int__ = sum_of_scores
Хммм, ако искаш това, защо просто не кръсти sum_of_scores като __int__?
+
+ def __lt__(self, second_instance):
+ return int(self) < int(second_instance)
+
+ def __le__(self, second_instance):
+ return int(self) <= int(second_instance)
+
+ def __gt__(self, second_instance):
+ return int(self) > int(second_instance)
+
+ def __ge__(self, second_instance):
+ return int(self) >= int(second_instance)
+
+ def __eq__(self, second_instance):
+ return int(self) == int(second_instance)
+
+ def __ne__(self, second_instance):
+ return int(self) != int(second_instance)
+
+ def __add__(self, second_instance):
+ return int(self) + int(second_instance)
+
+ def __sub__(self, second_instance):
+ return int(self) - int(second_instance)
Тук би било по-изчистено да използвам get_white_score и get_black_score, но реших да използвам предефинираните оператори
По-удачно би било, да. Освен това, добра идея е предварително да присвоиш в атрибут на инстанциите самите
ChessScoreобекти, за да не се налага да ги инициализираш при всяко използване на метод. Имайки ги като атрибути на инстанцията, просто ги сравняваш. Те самите вече са подготвени за сравнение, поради изискванията на задачата.Бих дефинирал един речник предварително, базиран на самия fen, за да мога тук просто да потърся ключ, съвпадащ с
pos. Реално този речник би ти свършил чудесна работа и за останалите методи.Сетих се малко късно за този тип речник, чак когато разписвах тази функция, но съм съгласна, че може да се дефинира по-рано и да се използва на други места.