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

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

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

输入格式

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

输出格式

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

样例输入

3 2
10 20 40

样例输出

10

数据规模和约定

　　N, M<=7

n, m = map(int, input().split())
nums = list(map(int, input().split()))
ans = 999999  # 初始答案设置一个比较大的数字，每次计算出一个极差就取较小值


def dfs(nums1):
    length = len(nums1)
    global ans
    # 当数组长度等于目标长度时记录新的极差，然后跳出当前深度
    if length == m:
        ans = min(ans, (max(nums1) - min(nums1)))
        return
    # 下面的嵌套循环是取出所有的可以合并的数字，并且替换成他们的和，再回溯，直到数组的长度为目标长度时跳出
    # 因为题目的数据范围较小，所以可以利用此方法暴力破解
    for i in range(length):  # 取出第一个数
        a = nums1[i]
        nums2 = nums1.copy()  # 每次操作都要对数组进行复制，以免影响下次的处理
        nums2.pop(i)
        length2 = len(nums2)
        for j in range(length2):  # 取出第二个数
            b = nums2[j]
            a += b
            nums3 = nums2.copy()
            nums3.pop(j)
            nums3.insert(i, a)  # 删掉a和b然后将a，b的和插入其中
            dfs(nums3)  # 回溯，加深深度，直到数组长度为目标长度
            a -= b  # 深层的回溯处理完后，要开始合并新的数，就要将a变回原来的值


dfs(nums)
print(ans)