棋盘放米问题代码_[算法系列]搞懂DFS(2)——模式套路+经典例题详解（n皇后问题，素数环问题）...

dfs(参数1,参数2,参数3,...):
    if 边界条件:
        //副作用,比如添加到解集中
        return
    从候选集中按照选优条件取出一个 e:
        if e 符合条件:
            这里可能改变状态,比如填上一个数字
            dfs(参数1,参数2,参数3,...) //这里某些参数可能会变,以表示到达下一个状态,比如dfs(arr,x+1,y+1)
            回溯,将刚刚的改变复位

'''
n 皇后问题

在n×n格的棋盘上放置彼此不受攻击的n个皇后。
按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
即n后问题等价于再n×n的棋盘上放置n个皇后，任何2个皇后不妨在同一行或同一列或同一斜线上。

输入 n, 返回解法的个数
'''

res = []#存放最终结果的list, 里面存放的每个正确结果的rec
def n_queue(n):
    rec = [-100] *n    #存放解, rec[i]表示第i行的皇后放在第rec[i]上
    dfs_n_queue(rec,0)    #从第0行开始

    print(res)

def dfs_n_queue(rec, row):
    if row == len(rec):     # 如果达到边界,即所有行都移填上
        rec_cp = rec.copy()
        res.append(rec_cp)  #res中加入rec的克隆
        return

    for y in range(0 , len(rec)):    #  对当前行的每一列进行试探
        if check(rec, row, y):       #   判断每行每列每斜线,若没有第二个,则check返回true
            rec[row] = y            #   检查成功,rec[row]=y 表示row+1行y+1列放入皇后
            dfs_n_queue(rec, row + 1)#状态转移到下一个位置
    rec[row ]= -100 # 均不合法,返回上一层前将该位置恢复

# 判断合法的函数    
def check(rec , x, y):
    for i in range(0 ,  len(rec)):
        #判断同行 如果行数相同,跳过
        if i == x :
            continue
        #判断同列: 
        if rec[i] == y:
            return  False
        #判断主对角线 比如 (0,0)和(1,1) =>横纵坐标差相等
        if rec[i] - i ==  y-x:
            return  False
        #判断副对角线 比如 (1,1)和(0,2) =>横纵坐标和相等
        if rec[i] + i == y + x:
            return  False
    return  True
n_queue(4)
# print: [[1, 3, 0, 2], [2, 0, 3, 1]]

'''
输入正整数n,对1-n进行排列, 使得相
输出时从整数1开始,逆时针排列,同一个环应该恰好输出1次邻两个数之和均为质数
n<=16

如输入:6
输出:
1 4 3 2 5 6 
1 6 5 2 3 4
'''

1,2     ...             [满足]
1,2,3   ...             [满足]
1,2,3   ...             [满足]
1,2,3,4 ...             [满足]
1,2,3,4,5 ...           [不满足,回溯]
1,2,3,4,6,...           [不满足,回溯]
1,2,3,4 ...             [没有更多选择,继续回溯]
1,2,3, ...              [没有更多选择,继续回溯]
1,2,5, ...
...

def zhishu_circle(n):
    rec = [0] * n       #定义状态数组, 即上述的[1,2,3...]
    rec[0] = 1
    dfs_zhishu_circle(n, rec, 1) # rec[0]固定为1,从下标为1开始


def dfs_zhishu_circle(n,rec,cur):
    if cur == n:    #当前行到达边界,所有数字填完,得到结果
        print(rec)
        return

    for tem in range(2 , n +1):     #对2~n+1按顺序进行选择试探
        if check_zhishu(n,rec,cur,tem): #如果满足题目要求
            rec[cur] = tem
            dfs_zhishu_circle(n, rec ,cur+1)
    rec[cur] = 0            # 均不满足, 返回时恢复

def check_zhishu(n , rec, cur,tem):     
    for i in range(0, cur):
        if rec[i] == tem :  #这个数字当然取两遍
            return False
    res_l = check_zhishu_core(rec[cur - 1] + tem) #判断左边和质数
    if (rec[(cur+1) % n] == 0):     #如果右边没填, 不考虑右边
        res_r = True
    else: #否则也要考虑右边相邻和
        res_r =check_zhishu_core(tem + rec[(cur + 1) % n])
    return res_l and res_r

import math
#判断k是否为质数
def check_zhishu_core(k):
    for i in range(2, k):
        if (k % i == 0):
            return False
    return True