python - 如何获得此组合列表?
问题描述
我有两个数字,N 和 L(比如说 5 和 3)。
如何生成list
列表之和等于 N (5) 且每个长度list
为 L (3) 的所有可能值?
示例输出(在这种情况下):
[0, 0, 5]
[0, 1, 4]
[0, 2, 3]
[0, 3, 2]
...
[0, 5, 0]
...
[1, 4, 0]
...
[5, 0, 0]
我已经检查了itertools
它的combinations
功能permutations
,但它们似乎不适合这项任务。
解决方案
您可以创建一个递归函数来生成具有给定条件的所有可能排列,然后过滤以仅保留总和为所需值的列表:
def list_results(a, b):
return [i for i in permutations(b) if sum(i) == a]
def permutations(d, current = []):
if len(current) == d:
yield current
else:
for i in range(10):
yield from permutations(d, current+[i])
print(list_results(5, 3))
输出:
[[0, 0, 5], [0, 1, 4], [0, 2, 3], [0, 3, 2], [0, 4, 1], [0, 5, 0], [1, 0, 4], [1, 1, 3], [1, 2, 2], [1, 3, 1], [1, 4, 0], [2, 0, 3], [2, 1, 2], [2, 2, 1], [2, 3, 0], [3, 0, 2], [3, 1, 1], [3, 2, 0], [4, 0, 1], [4, 1, 0], [5, 0, 0]]
编辑:稍微快一点将需要对递归函数进行额外检查:
import time
def timeit(f):
def wrapper(*args, **kwargs):
c = time.time()
results = list(f(*args, **kwargs))
print("Result from function '{}' achieved in {}".format(f.__name__, abs(c-time.time())))
return results
return wrapper
@timeit
def outer_permutations():
def permutations1(d, b, current = []):
if len(current) == d:
yield current
else:
for i in range(10):
if len(current) < 2 or sum(current+[i]) == b:
yield from permutations1(d, b, current+[i])
yield from permutations1(3, 5)
@timeit
def list_results(a, b):
return [i for i in permutations(b) if sum(i) == a]
v = outer_permutations()
v1 = list_results(3, 5)
输出:
Result from function 'outer_permutations' achieved in 0.0006079673767089844
Result from function 'list_results' achieved in 0.09148788452148438
请注意,这两个函数的输出是:
[[0, 0, 5], [0, 1, 4], [0, 2, 3], [0, 3, 2], [0, 4, 1], [0, 5, 0], [1, 0, 4], [1, 1, 3], [1, 2, 2], [1, 3, 1], [1, 4, 0], [2, 0, 3], [2, 1, 2], [2, 2, 1], [2, 3, 0], [3, 0, 2], [3, 1, 1], [3, 2, 0], [4, 0, 1], [4, 1, 0], [5, 0, 0]]
推荐阅读
- javascript - javascript匿名函数的生命周期是多少?
- javascript - 使用映射和索引 es6 更新嵌套数组值
- c# - 带参数的 ICommand 接口不起作用
- ruby-on-rails - Ruby on Rails 搜索方法
- django - Celery - 权限问题 - 创建文件夹
- mysql - 如何在 Alpine Docker 中编译 MySQL 5.7
- javascript - 如何像在二维平面上使用“相机”一样为页面设置动画,特别是平移和缩放?
- c++ - 当用作函数参数时,C++“忘记”该变量是 constexpr
- python - Scrappy 不抓取 div 类中的跨度文本
- ruby - 拆分字符串以仅获取前 5 个字符