100127. 给小朋友们分糖果 II

给你两个正整数 nlimit

请你将 n 颗糖果分给 3 位小朋友,确保没有任何小朋友得到超过 limit 颗糖果,请你返回满足此条件下的 总方案数

示例 1:

复制代码
输入:n = 5, limit = 2
输出:3
解释:总共有 3 种方法分配 5 颗糖果,且每位小朋友的糖果数不超过 2 :(1, 2, 2) ,(2, 1, 2) 和 (2, 2, 1) 。

示例 2:

复制代码
输入:n = 3, limit = 3
输出:10
解释:总共有 10 种方法分配 3 颗糖果,且每位小朋友的糖果数不超过 3 :(0, 0, 3) ,(0, 1, 2) ,(0, 2, 1) ,(0, 3, 0) ,(1, 0, 2) ,(1, 1, 1) ,(1, 2, 0) ,(2, 0, 1) ,(2, 1, 0) 和 (3, 0, 0) 。

解析:

正难则反:我们先算出总方案数,在减去不合法的数即可。

有组合数学可以用隔板法进行求出总方案数。C(n+2,2)

在考虑不合法的数。

由容斥原理得:

考虑到三种情况:(这里要注意至少这个词语的理解)

1.当至少一个人不合法,那个人至少要limit+1个球,在用隔板法,在n-limit-1中经行分配 3*C(n- limit-1+2,2),由题意知道由三个人。

2.当至少有两个人不合法时,至少要消耗(limit+1)*2个球,分配的方案数为3⋅C(n−2⋅(limit+1)+2,2)

3.至少有三个人不合法时,分配的方案数为C(n−3⋅(limit+1)+2,2)

最后总方案数 - 不合法方案数即可。

在上面三种情况中有一些是重复被减去 所以后面要加上。

复制代码
class Solution {
public:
    long long c2(long long n){
        return n > 1 ?n*(n-1)/2:0;
    }
    long long distributeCandies(int n, int limit) {
        return c2(n+2) - 3*c2(n-limit+1) + 3*c2(n-2 *limit) - c2(n-3*limit - 1);
    }
};
相关推荐
江畔柳前堤5 小时前
大语言模型分布式训练:从并行策略到万卡工程的系统梳理
人工智能·分布式·深度学习·算法·目标检测·机器学习·语言模型
Forever Nore6 小时前
LeetCode 4 寻找两个正序数组的中位数 - 二分
算法·leetcode
罗西的思考8 小时前
【OpenClaw具身硬件】MiniClaw 阅读笔记---(1)基础
人工智能·算法·机器学习
蛋先生DX9 小时前
大模型参数存储格式揭秘:BF不是男朋友
深度学习·算法·llm
爱跳舞的烤冷面9 小时前
自学嵌入式第22天(数据结构——哈希)
数据结构·算法·哈希算法
猎嘤一号9 小时前
博弈论(Game Theory)的理论、算法与工程
人工智能·算法·安全·博弈论
月光船幽幽10 小时前
影子模式下保护 logits 不被修改
人工智能·python·算法
马拉AI10 小时前
腾讯开源 Agent 记忆系统,AI“换对话就忘”的问题有了新解法(附安装使用教程)
人工智能·算法·开源·科研
地平线开发者11 小时前
征程6工具链模型X86推理方式说明
算法
地平线开发者11 小时前
【征程6】校准量化中HistogramObserver解析
算法