Python课程设计 第 15 / 52 篇

迭代生成-参考答案

WechatIMG12.jpeg

def conflict(chesses_placed, next_chess_xcor):
    """
    This is a conflicting function defined
    :param chesses_placed: use a tuple to save chesses placed
    :param next_chess_xcor: the x coordinate of next chess to place
    :return: true or false
    """
    # get the next y coordinate by chessed placed
    next_chess_ycor = len(chesses_placed)

    # use next chess y coordinate to confirm the count of loop
    for i in range(next_chess_ycor):

        # use absolute value to confirm the same oblique line
        if abs(chesses_placed[i] - next_chess_xcor) in (0, abs(i - next_chess_ycor)):
            return True
    return False

def place_queens(num=8, chesses_placed_tuple=()):
    """
    This is a placing queens function defined
    :param num: the count of queens
    :param chesses_placed_tuple: such as (1, 5, 0)
    :return: a generator
    """
    for i in range(num):
        if not conflict(chesses_placed_tuple, i):

            # judge the last line
            if len(chesses_placed_tuple) == num - 1:
                yield (i,)
            else:

                # get one bye one element from the generator
                for j in place_queens(num, chesses_placed_tuple + (i,)):
                    # superimpose elements to the newest tuple
                    yield (i,) + j

res = list(place_queens(8))
print(res)
print(len(res))