Question

我正在尝试使用位操作来查找功率集。我可以生成所有集，但它们不可索引。我无法将其保存为列表清单。我试图通过互联网找到解决方案但无法获得相关信息。

这是我使用的代码。

n = int(input()) # Size of the array
noteValue = []   # Array whose power set is to be found
for i in range(n):
    noteValue.append(int(input()))


powerSet = []
for i in range(1<<n):
    for j in range(n):
        if (i & (1<<j) > 0 ):
            powerSet.append(noteValue[j])

print(powerSet)

输出：

[1, 2, 1, 2, 3, 1, 3, 2, 3, 1, 2, 3]

期望的输出：

[[], [1], [2], [1, 2], [3], [1, 3], [2, 3], [1, 2, 3]]

Answer 1

实际上，您可以使用此示例中的临时列表sub：

powerSet = []
for i in range(1<<n):
    # Add sub list 
    sub = []
    for j in range(n):
        if (i & (1<<j) > 0 ):
            # Append to sub list
            sub.append(noteValue[j])
    # Then append sub to pwerset after finishing the inner loop
    powerSet.append(sub)

print(powerSet)

所以，通过这个输入：

2
2
3

将输出：

[[], [2], [3], [2, 3]]

Python：在不使用任何内置函数（如itertools等）的情况下查找功率集

1 个答案: