打卡记录
给小朋友们分糖果 II(容斥原理)
链接
大佬的题解
def c2(n: int) -> int:
return n * (n - 1) // 2 if n > 1 else 0
class Solution:
def distributeCandies(self, n: int, limit: int) -> int:
return c2(n + 2) - 3 * c2(n - limit + 1) + 3 * c2(n - 2 * limit) - c2(n - 3 * limit - 1)