本文介绍一种高效生成“带符号幂集”的方法：对输入序列中每个元素，独立选择保留原值、取负值或完全排除，从而生成所有合法子集，避免后续过滤，时间复杂度最优为 o(3ⁿ)。
本文介绍一种高效生成“带符号幂集”的方法：对输入序列中每个元素，独立选择保留原值、取负值或完全排除，从而生成所有合法子集，避免后续过滤，时间复杂度最优为 o(3ⁿ)。
在标准幂集（powerset）中，每个元素仅有“包含”或“不包含”两种状态；而本问题要求每个元素具备三种互斥状态：保持原值（+1）、取负值（−1）或不参与（0）。这本质上是构建一个长度为 n 的三元笛卡尔积，而非二元子集枚举——因此直接使用 itertools.product([-1, 0, 1], repeat=n) 是最自然且最优的解法。
以下为完整实现：
import itertools
def extended_powerset(elements):
elements = list(elements)
n = len(elements)
for option in itertools.product([-1, 0, 1], repeat=n):
# 对每个元素按对应符号系数处理：-1→取负，0→跳过，1→保留
subset = [coeff * elem for coeff, elem in zip(option, elements) if coeff != 0]
yield subset使用示例：
elements = [-2, 1] result = list(extended_powerset(elements)) print([sorted(s) for s in result]) # 可选排序以便验证（因顺序无关） # 输出（共 3² = 9 个子集）： # [[], [-2], [2], [-1], [1], [-2, -1], [-2, 1], [2, -1], [2, 1]]
✅ 优势说明：
- 零冗余：不生成非法组合（如 [−1, 1] 或 [−2, 2]），无需后置过滤；
- 时间最优：精确生成全部 3ⁿ 个合法子集，无浪费迭代；
- 通用性强：适用于任意可乘标量的元素类型（如 int, float, 甚至支持 __mul__ 的自定义对象）；
- 内存友好：返回生成器，适合大规模输入（可通过 list(...) 强制展开，或逐项消费）。
⚠️ 注意事项：
- 输入中若含 0，需注意 −0 == 0，此时 +0 与 −0 在结果中不可区分（Python 中二者均为 0），若业务需区分符号零，应改用 decimal.Decimal 或自定义包装；
- 若需固定输出顺序（如按字典序或子集大小），可在最终结果上调用 sorted()，但生成过程本身不保证顺序；
- 该算法不依赖元素唯一性假设，但题目已声明“无重复”，故无需额外去重逻辑。
综上，itertools.product 驱动的三态选择方案，是以清晰语义、最小计算开销解决此类“带符号子集枚举”问题的标准范式。
