【LeetCode：2928】给小朋友们分糖果 I（Java）

题目链接

• 2928. 给小朋友们分糖果 I

题目描述

给你两个正整数 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) 。

提示：

1 <= n <= 50
1 <= limit <= 50

求解思路1

• 暴力枚举：i和j表示第一个和第二个小朋友分得糖果数，设置边界条件，判断第三个小朋友是否满足limit。

实现代码1

class Solution {
    public int distributeCandies(int n, int limit) {
        int ans = 0;
        for (int i = 0; i <= limit; i ++) {
            for (int j = 0; j <= limit; j ++) {
                if (i + j > n) {
                    break;
                }
                if (n - i - j <= limit) {
                    ans++;
                }
            }
        }
        return ans;
    }
}

求解思路2

• 枚举优化：当第一个小朋友分的i个糖果时（这里需要限制i不会超过n和limit中小的那个），剩下的糖果数量n-i应当不超过给第二个和第三个小朋友的限制，如果超过了直接continue，第二个小朋友最多拿min(limit, n - i)个，最少拿max(0, n - i - limit)个，因此共有Math.min(limit, n - i) - Math.max(0, n - i - limit) + 1种方案。

实现代码2

class Solution {
    public int distributeCandies(int n, int limit) {
        int ans = 0;
        for (int i = 0; i <= Math.min(limit, n); i ++) {
            if (n - i > 2 * limit) {
                continue;
            } else {
                ans += Math.min(limit, n - i) - Math.max(0, n - i - limit) + 1;
            }
        }
        return ans;
    }
}