常见面试算法题-叠积木

■ 题目描述

有一堆长方体积木，它们的长度和宽度都相同，但长度不一。

小橙想把这堆积木叠成一面墙，墙的每层可以放一个积木，也可以将两个积木拼接起来，要求每层的长度相同。

若必须用完这些积木，叠成的墙最多为多少层？

如下是叠成的一面墙的图示，积木仅按宽和高所在的面进行拼接。

输入描述：

输入为一行，为各个积木的长度，数字为正整数，并由空格分隔。积木的数量和长度都不超过5000。

输出描述：

输出一个数字，为墙的最大层数，如果无法按要求叠成每层长度一致的墙，则输出-1。

示例1   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

3 6 3 3 3

输出

3

解释：以 6 为底的墙，第一层为 6 ，第二层为 3 + 3，第三层 3 + 3。

示例2   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

9 9 9 5 3 2 2 2 2 2

输出

5

解释：

5+2+2=9

3+2+2+2=9

9,9,9

共五层

以下代码为本人原创，可以供大家参考，若有不足之处，感谢指出！！！！

nums = list(map(int, input().split()))
max_nums = max(nums)
sum_nums = sum(nums)
k = sum_nums//max_nums


def check(nums, k):
    def dfs(state, summ):
        if state == (1 << n)-1:
            return True
        for j in range(n):
            if summ + nums[j] > target:
                break
            if state & (1 << j) == 0:
                next_state = state + (1 << j)
                if dfs(next_state, (summ + nums[j]) % target):
                    return True
        return False
    if sum_nums%k != 0:
        return False
    n = len(nums)
    target = sum_nums//k
    nums.sort()
    if nums[-1] > target:
        return False
    return dfs(0, 0)


for i in range(k, 0, -1):
    if check(nums, i):
        print(i)
        break
    if i == 1:
        print(-1)