100127. 给小朋友们分糖果 II

给你两个正整数 n 和 limit 。

请你将 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);
    }
};
相关推荐
All for pursuit.25 分钟前
【回溯-7】494.目标和
数据结构·c++·算法·leetcode
白杨尚青26 分钟前
C++入门篇(十三):vector(上)——动态数组:构造、空间增长与迭代器失效
开发语言·c++·笔记·算法·stl
Jasmine_llq27 分钟前
《P17461 [GESP202609 八级] 生成树计数》
算法·栈·tarjan 算法·边双连通分量·深度优先搜索 dfs·tarjan 保存当前路径节点·模乘法
Jasmine_llq29 分钟前
《P17462 [GESP202609 八级] 末班车》
算法·dijkstra 算法·最大化版本,求最长可行值·建反向图(逆图 / 反转边)·多源 dijkstra 思想·对每个终点单独跑一遍djst·离线预处理 + 在线查询
the3clipse1 小时前
H.265熵编码核心:CABAC自适应二进制算术编码详解——如何将语法元素高效压缩为比特流
人工智能·算法·视频编码·h.265·hevc·cabac·cavlc
CQU_JIAKE1 小时前
10.4【A】
算法
Godspeed Zhao2 小时前
现代智能雷达技术19——算法(5)
算法
(Charon)2 小时前
【C++面试】RAII到底是什么:从资源管理理解C++对象生命周期
c++·算法·面试
Omics Pro2 小时前
之江实验室NAR|虚拟细胞3阶段训练范式
数据库·人工智能·算法·机器学习·自然语言处理
Tisfy2 小时前
LeetCode 0678.有效的括号字符串:O(n)+O(1)一次遍历
算法·leetcode·字符串·题解·贪心·括号匹配