0

0

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

花韻仙語

花韻仙語

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

|

154人浏览过

|

来源于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 可精确去重,确保每个唯一元组仅出现一次。

以下是完整实现:

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)

使用示例

Pebblely
Pebblely

AI产品图精美背景添加

下载
# 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创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

344

2023.10.09

更新pip版本
更新pip版本

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

420

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、保存并关闭文件即可。

775

2024.12.23

python升级pip
python升级pip

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

353

2025.07.23

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

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

428

2023.08.14

1688阿里巴巴货源平台入口与批发采购指南
1688阿里巴巴货源平台入口与批发采购指南

本专题整理了1688阿里巴巴批发进货平台的最新入口地址与在线采购指南,帮助用户快速找到官方网站入口,了解如何进行批发采购、货源选择以及厂家直销等功能,提升采购效率与平台使用体验。

48

2026.02.06

快手网页版入口与电脑端使用指南 快手官方短视频观看入口
快手网页版入口与电脑端使用指南 快手官方短视频观看入口

本专题汇总了快手网页版的最新入口地址和电脑版使用方法,详细提供快手官网直接访问链接、网页端操作教程,以及如何无需下载安装直接观看短视频的方式,帮助用户轻松浏览和观看快手短视频内容。

6

2026.02.06

C# 多线程与异步编程
C# 多线程与异步编程

本专题深入讲解 C# 中多线程与异步编程的核心概念与实战技巧,包括线程池管理、Task 类的使用、async/await 异步编程模式、并发控制与线程同步、死锁与竞态条件的解决方案。通过实际项目,帮助开发者掌握 如何在 C# 中构建高并发、低延迟的异步系统,提升应用性能和响应速度。

5

2026.02.06

Python 微服务架构与 FastAPI 框架
Python 微服务架构与 FastAPI 框架

本专题系统讲解 Python 微服务架构设计与 FastAPI 框架应用,涵盖 FastAPI 的快速开发、路由与依赖注入、数据模型验证、API 文档自动生成、OAuth2 与 JWT 身份验证、异步支持、部署与扩展等。通过实际案例,帮助学习者掌握 使用 FastAPI 构建高效、可扩展的微服务应用,提高服务响应速度与系统可维护性。

4

2026.02.06

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.4万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号