LC2928给小朋友们分糖果

题目

只有3个小朋友，限制每个小朋友分到不超过limit个糖果，总共有n个糖，分完的方法有几种？

等价问题

插入两个隔板，将n分为三块，每块大小不超过limit，的分法有几种

解题

解法1：递归，相当于枚举

class Solution {
public:
    int res = 0;
    int distributeCandies(int n, int limit) {
        if (3 * limit < n)
            return 0;
        iterate(2, 0, limit, n);
        return res;
    }
    void iterate(int k, int now, int limit, int n) {
        if (k == 0) {
            if ((n - now) <= limit)
                res++;
            return;
        }
        for(int i = now; i <= n && (i-now) <= limit; ++i){
            //！！要等于号，因为最后一个人可以不分
            iterate(k-1, i, limit, n);
        }
    }
};

复杂度

• 时间：O(limit*limit)
• 空间：O(1)递归深度是3

解法2：枚举优化

第一个人分x，还剩n-x

• n-x > limit*2: 不会满足，直接跳过
• n-x <= limit*2: 第二个小朋友至少要分 max(0, n-x-limit)个，一共有min(limit, n-x)种分法，所以在第一个人分完后，剩下的情况为min(limit, n-x) - max(0, n-x-limit) + 1.

class Solution {
public:
    int distributeCandies(int n, int limit) {
        if (3 * limit < n)
            return 0;
        int res = 0;
        for(int i = 0; i <= limit && i <= n; ++i){
            //！！要等于号，因为第一个人可以分完
            if(n - i > 2 * limit) continue;
            res += min(limit, n-i) - max(0, n-i-limit) + 1;
        }
        return res;
    }
};

复杂度

• 时间：O(min(limit,n))
• 空间：O(1)