讲师中心 微信公众号
AI工具推荐 视频效率加速

生成所有含 d 个非零元(取值为 ±val)的 r-元组

冬宇小哥_4050

冬宇小哥_4050

发布时间:2026-02-06 12:48:39

|

174人浏览过

|

来源于php中文网

原创

生成所有含 d 个非零元(取值为 ±val)的 r-元组

本文介绍一种高效、可扩展的方法,使用 `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 可精确去重,确保每个唯一元组仅出现一次。

以下是完整实现:

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
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)

✅ 使用示例:

# d=2, r=5, val=7 → 共 C(5,2) × 2² = 10 × 4 = 40 个元组
list(construct_r_tuples(7, 2, 5))[:5]  # 前5个
# [(-7, -7, 0, 0, 0), (-7, 0, -7, 0, 0), (-7, 0, 0, -7, 0), (-7, 0, 0, 0, -7), (0, -7, -7, 0, 0)]

# d=1, r=3 → 共 C(3,1) × 2¹ = 3 × 2 = 6 个
list(construct_r_tuples(7, 1, 3))
# [(-7, 0, 0), (0, -7, 0), (0, 0, -7), (0, 0, 7), (0, 7, 0), (7, 0, 0)]

⚠️ 注意事项:

  • more-itertools 需单独安装:pip install more-itertools;
  • 输出为生成器(yield),内存友好;若需全部加载进内存,请显式调用 list(...);
  • 总数量为 $\binom{r}{d} \cdot 2^d$,随 r 和 d 增长极快(如 r=10, d=4 已达 5040 个),建议按需迭代而非全量缓存;
  • 若 d=0,直接返回全零元组;若 d>r,函数自动返回空迭代器。

该方法将组合逻辑解耦为「符号选择」与「位置分配」两个正交步骤,结构清晰、易于测试与扩展(例如支持多值非零集 {±val₁, ±val₂} 只需修改 combinations_with_replacement 的输入)。对于算法原型、组合搜索或教学演示,是比手工嵌套循环更健壮、更 Pythonic 的选择。

热门AI工具

更多
DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

相关专题

更多
pip安装使用方法
pip安装使用方法

安装步骤:1、确保Python已经正确安装在您的计算机上;2、下载“get-pip.py”脚本;3、按下Win + R键,然后输入cmd并按下Enter键来打开命令行窗口;4、在命令行窗口中,使用cd命令切换到“get-pip.py”所在的目录;5、执行安装命令;6、验证安装结果即可。大家可以访问本专题下的文章,了解pip安装使用方法的更多内容。

3490

2023.10.09

更新pip版本
更新pip版本

更新pip版本方法有使用pip自身更新、使用操作系统自带的包管理工具、使用python包管理工具、手动安装最新版本。想了解更多相关的内容,请阅读专题下面的文章。

7923

2024.12.20

pip设置清华源
pip设置清华源

设置方法:1、打开终端或命令提示符窗口;2、运行“touch ~/.pip/pip.conf”命令创建一个名为pip的配置文件;3、打开pip.conf文件,然后添加“[global];index-url = https://pypi.tuna.tsinghua.edu.cn/simple”内容,这将把pip的镜像源设置为清华大学的镜像源;4、保存并关闭文件即可。

9622

2024.12.23

python升级pip
python升级pip

本专题整合了python升级pip相关教程,阅读下面的文章了解更多详细内容。

5047

2025.07.23

页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

6856

2023.08.14

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

20

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

20

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

20

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

40

2026.11.05

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn