蓝桥杯算法训练-粘木棍（Python）

问题描述

　　有N根木棍，需要将其粘贴成M个长木棍，使得最长的和最短的的差距最小。

输入格式

　　第一行两个整数N,M。
　　一行N个整数，表示木棍的长度。

输出格式

　　一行一个整数，表示最小的差距

样例输入

3 2
10 20 40

样例输出

10

数据规模和约定

　　N, M<=7

---

1.我的思路：

        这道题涉及到了深度优先搜索算法，递归与回溯的知识。让列表中任意的两个元素a和b相加得到c，再从列表中删除元素a和元素b的同时 加入元素c，重复这样的操作，直到列表长度为m。

2.什么是深度优先算法？

       学习深度优先算法https://blog.csdn.net/raphealguo/article/details/7560918?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522164750612016782089379686%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=164750612016782089379686&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-7560918.142%5Ev2%5Ees_vector_control_group,143%5Ev4%5Econtrol&utm_term=%E6%B7%B1%E5%BA%A6%E4%BC%98%E5%85%88%E7%AE%97%E6%B3%95&spm=1018.2226.3001.4187

---

100分代码如下：

# 2022.3.16
# 蓝桥杯算法训练-粘木棍
# 知识点：深度优先搜索算法
# 我的思路：让列表中任意的两个元素a和b相加得到c，再从列表中删除元素a和元素b的同时
# 加入元素c，重复这样的操作，直到列表长度为m。为此我使用深度优先搜索算法，其中
# 包含了递归与回溯的知识。

n, m = list(map(int, input().split()))
nums = list(map(int, input().split()))
results = []

def dfs(nums):
    global m, results
    lenth = len(nums)
    if lenth == m:       # 递归出口
        #print(nums)
        x = max(nums) - min(nums)
        results.append(x)
        return
    for i in range(lenth):
        a = nums[i]
        nums2 = nums.copy()  # 须在新的列表上进行操作，否则会影响i的下一次循环
        nums2.pop(i)         # 选择i与嵌套循环中的j相加，且删除i和j
        lenth2 = len(nums2)
        for j in range(lenth2): # 遍历删除i之后的列表
            b = nums2[j]
            a = a + b
            nums3 = nums2.copy()  # 为了避免影响j的下一次循环
            nums3.pop(j)
            nums3.insert(i, a)
            dfs(nums3)  # 递归入口
            a = a - b   # 回溯
dfs(nums)        
print(min(results))

提交结果：