2N皇后

# 2n queens
def solve(row):
    # 传入一个行数就够了,map_lst 是可变数组,可以直接修改之,
    # 计数的ans设置为全局变量,列数每一次都要从0访问到n-1
    global ans
    if row == 2 * n:
        ans += 1
        return 0  # return几无所谓,我们只是想要计数,计完数停止
    if row < n:  # 如果放黑棋
        put_in = 2
    else:
        put_in = 3
    real_row = row % n  # 下面用real_row来进行访问
    for tmp_col in range(0, n):  # 当前行,所有列都要考虑
        if is_place(row, tmp_col):  # 如果可以放,就放
            map_lst[real_row][tmp_col] = put_in
            solve(row + 1)  # 放完,进行递归,考虑下一行
            map_lst[real_row][tmp_col] = 1  # 上一层递归要么摆放成功,ans加 1;要么不成功,不用操作,直接return停止。不管是那种情况
            # 我们都要回溯,继续考虑这一行中的下一个
    return 0  # 结束


def is_place(row, col):  # row表示行,col表示列
    if row < n:  # 模为n,即如果放了前面0-n-1行(黑棋放好),row%n得0,从第0行开始放白棋
        judge_sign = 2  # 如果在放黑棋,而黑棋用2表示,所以接下来的判断标志设为2
    else:
        judge_sign = 3  # 否则为白棋
    # 白皇后黑皇后统称为棋子
    # 四种情况,1、当前位置能否放、有无别的棋子 2、当前列有无冲突 3、左上对角线 4、右上对角线
    row = row % n
    if map_lst[row][col] != 1:  # 可能等于0,也可能放了另一个颜色的棋子
        return False
    for tmp_row in range(row - 1, -1, -1):  # 判断列
        if map_lst[tmp_row][col] == judge_sign:
            return False
    tmp_row = row - 1
    tmp_col = col - 1
    while tmp_row >= 0 and tmp_col >= 0:  # 判断左上对角线
        if map_lst[tmp_row][tmp_col] == judge_sign:
            return False
        tmp_row -= 1
        tmp_col -= 1
    tmp_row = row - 1
    tmp_col = col + 1
    while tmp_row >= 0 and tmp_col <= n - 1:  # 判断右上对角线
        if map_lst[tmp_row][tmp_col] == judge_sign:
            return False
        tmp_row -= 1
        tmp_col += 1
    return True


if __name__ == "__main__":
    n = int(input())  # 输入的规格
    map_lst = []  # 地图
    for i in range(n):
        one_line = [int(j) for j in input().split()]
        map_lst.append(one_line)
    ans = 0  # 计数的答案
    solve(0)
    print(ans)

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值