本文介绍一种高效生成“扩展幂集”的方法：对输入可迭代对象中的每个元素，允许其以正值、负值或完全不出现于子集中——共三种独立选择，避免生成冗余组合后再过滤。
本文介绍一种高效生成“扩展幂集”的方法：对输入可迭代对象中的每个元素，允许其以正值、负值或完全不出现于子集中——共三种独立选择，避免生成冗余组合后再过滤。
在标准幂集（powerset）中，每个元素仅有“包含”或“不包含”两种状态，对应 $2^n$ 个子集。而本问题要求更丰富的语义：每个元素 $x$ 可表现为 $x$、$-x$ 或完全缺席，但不可同时出现 $x$ 和 $-x$，也不可出现 $x$ 与其原始符号冲突的副本（如原列表含 -2，则 2 是其“正向表示”，而非新元素）。这本质上是为每个元素赋予三个互斥选项：+1（取正值）、-1（取负值）、0（排除）。
该问题恰好映射到长度为 $n$ 的三元笛卡尔积：${-1, 0, 1}^n$，共 $3^n$ 种组合。使用 itertools.product 可直接、惰性地遍历所有合法状态，无需构造超集再过滤，时间与空间效率均为最优。
以下为完整实现：
import itertools
def extended_powerset(elements):
"""
生成扩展幂集：每个元素可取正值、负值或不出现。
Args:
elements: 可迭代对象（如 list, tuple），元素应支持一元负号运算（如 int, float）
Yields:
list: 每个合法子集（顺序与输入一致，内部无序性由调用方保证）
"""
elements = list(elements)
n = len(elements)
# 为每个位置生成 {-1, 0, 1} 的选择
for signs in itertools.product((-1, 0, 1), repeat=n):
subset = []
for sign, elem in zip(signs, elements):
if sign != 0:
subset.append(sign * elem)
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] 等非法组合；
- 惰性求值：yield 实现内存友好，适合大规模输入；
- 通用性强：支持任意支持 __neg__ 的类型（如 int, float, Fraction, 自定义数值类）；
- 可预测顺序：按三元笛卡尔积字典序生成，便于调试与测试。
⚠️ 注意事项：
- 若输入含 0，注意 +0 与 -0 在 Python 中相等（0 == -0 is True），将导致 [0] 和 [-0] 实际相同；若需区分，应预处理或改用 sign 函数控制；
- 该算法不保证子集内部排序（如 [2, -1] 与 [-1, 2] 视为同一集合），若需标准化输出，可在 yield 前对 subset 排序（如 yield sorted(subset)）；
- 复杂度为 $O(3^n)$，属指数级，适用于 $n \leq 20$ 左右；对更大规模，需考虑采样或增量生成策略。
综上，利用 itertools.product 构建三元选择空间，是从根本上契合问题语义的简洁、正确且高效的解决方案。
