生成所有含 d 个非零元（取值为 ±val）的 r-元组-Python教程

本文介绍一种高效、可扩展的方法，使用 `itertools` 和 `more-itertools` 构造所有长度为 `r`、恰好含 `d` 个非零元素（每个为 `+val` 或 `-val`）、其余为 `0` 的元组，避免嵌套循环与重复逻辑。

在组合构造任务中，当需要生成固定长度、指定非零元素个数及符号自由度的元组时（例如信号处理中的稀疏模式枚举、优化问题的候选解生成），硬编码多重循环不仅难以维护，更无法泛化到任意 d。理想方案应具备清晰语义、组合可证明性、时间复杂度可控三大特性。

核心思路分两步完成：

1. 生成所有 d 元符号组合：从集合 {+val, -val} 中允许重复地选取 d 个元素（顺序暂不重要），对应 itertools.combinations_with_replacement；
2. 将符号组合与 (r−d) 个零混合并枚举所有互异排列：由于零之间不可区分，直接使用 itertools.permutations 会产生大量重复；改用 more_itertools.distinct_permutations 可精确去重，确保每个唯一元组仅出现一次。

以下是完整实现：

from itertools import combinations_with_replacement, chain
from more_itertools import distinct_permutations

def construct_r_tuples(val, d, r):
    """
    构造所有长度为 r、恰好含 d 个非零元素（取 ±val）、其余为 0 的元组。

    Args:
        val (int/float): 非零元素的绝对值
        d (int): 非零元素个数（0 ≤ d ≤ r）
        r (int): 元组总长度

    Yields:
        tuple: 满足条件的 r 元组（如 (-7, 0, 7, 0, 0)）
    """
    if not (0 <= d <= r):
        return
    if d == 0:
        yield (0,) * r
        return

    # Step 1: 所有 d 元符号组合（允许重复，如 (+val,+val), (+val,-val), (-val,-val)）
    for signs in combinations_with_replacement((-val, +val), d):
        # Step 2: 补齐 r-d 个零，生成所有互异排列
        full_tuple = signs + (0,) * (r - d)
        yield from distinct_permutations(full_tuple)

✅ 使用示例：