Python: N-Queen拼图检查对角线的解决方案

def is_safe(board, row, col, N):
    # 检查当前位置的列是否安全
    for i in range(row):
        if board[i][col] == 1:
            return False

    # 检查左上对角线是否安全
    i = row - 1
    j = col - 1
    while i >= 0 and j >= 0:
        if board[i][j] == 1:
            return False
        i -= 1
        j -= 1

    # 检查右上对角线是否安全
    i = row - 1
    j = col + 1
    while i >= 0 and j < N:
        if board[i][j] == 1:
            return False
        i -= 1
        j += 1

    return True

def solve_n_queen(N):
    board = [[0] * N for _ in range(N)]

    def backtrack(row):
        if row == N:
            # 打印解决方案
            for i in range(N):
                for j in range(N):
                    print(board[i][j], end=' ')
                print()
            print()
        else:
            for col in range(N):
                if is_safe(board, row, col, N):
                    board[row][col] = 1
                    backtrack(row + 1)
                    board[row][col] = 0

    backtrack(0)

# 测试
solve_n_queen(4)