【算法之Python篇】回溯算法

最新推荐文章于 2024-03-10 13:23:39 发布

IsQiya

最新推荐文章于 2024-03-10 13:23:39 发布

阅读量330

点赞数

分类专栏： python笔记文章标签：算法 python leetcode

本文链接：https://blog.csdn.net/qq_45351292/article/details/121546537

版权

python笔记专栏收录该内容

45 篇文章 11 订阅

订阅专栏

本文介绍了如何使用Python实现经典的N皇后问题和全排列问题。N皇后问题中，通过检查每一行的皇后位置是否冲突，递归地放置皇后。全排列问题则利用回溯法生成所有可能的排列组合。这两个算法都涉及到回溯和递归策略，对于理解计算机科学中的搜索和优化问题具有重要意义。

摘要由CSDN通过智能技术生成

N皇后问题

def check_box(row_index):#检查0行到当前行是否符合要求的
    for i in range(row_index):
        if cols[i]==cols[row_index]:#检查i行和row_index行是否在同一列
            return False
        if row_index-i==abs(cols[i]-cols[row_index]):#检查i行和row_index行是否在同一斜线
            return False
    else:#循环奖励语句
        return True
        
def set_queen(row_index):
    if row_index==queens_num: #回溯中断
        res.append(cols[:]) #加入一个解决方案
        return
    for i in range(queens_num):
        cols[row_index] = i #当前行关键语句
        if check_box(row_index):#检查满足要求的皇后
            set_queen(row_index+1) #排好这一行就进行下一行
            
queens_num=4
res=[]#记录全部解决方案
cols=[0 for _ in range(queens_num)]  #”_“获取最后一次变量
set_queen(0)
print(res)


>>>[[1, 3, 0, 2], [2, 0, 3, 1]]

全排列

def permute(nums, m, solution=[]):
    if m == 0:  #回溯终止条件
        res.append(solution)
        return
    
    for i in range(len(nums)):
        new_solution = solution + [nums[i]]
        new_list = nums[:i] + nums[i+1:]  #把除了nums[i]（）也就是已经排列的组合去掉）的其它数合并
        permute(new_list, m-1, new_solution)
    return


res = []  #存储最终结果
permute([1,2,3,4], 3)
print(res)

>>>[[1, 2, 3], [1, 2, 4], [1, 3, 2], [1, 3, 4], [1, 4, 2], [1, 4, 3], [2, 1, 3], [2, 1, 4], [2, 3, 1], [2, 3, 4], [2, 4, 1], [2, 4, 3], [3, 1, 2], [3, 1, 4], [3, 2, 1], [3, 2, 4], [3, 4, 1], [3, 4, 2], [4, 1, 2], [4, 1, 3], [4, 2, 1], [4, 2, 3], [4, 3, 1], [4, 3, 2]]