LeetCode-1103. Distribute Candies to People（分糖果 II）

分糖果 II

Distribute Candies to People

解

方法一：暴力法
遍历一遍，这里有一点小优化，即假设最后一名正好分完，最后再把多的糖果减去，这样少了一行判断

class Solution {
public:
    vector<int> distributeCandies(int candies, int num_people) {
        int i = 0;
        vector<int>ret(num_people, 0);
        while (candies > 0) {
            ret[i % num_people] += i + 1;
            candies -= i + 1;
            i++;
        }
        ret[(i - 1) % num_people] += candies;

        return ret;
    }
};

方法二：等差数列
除了最后一份以外，其他礼物的数量都是从 1 开始构成的等差数列。
假设有complete份完整礼物，糖果总数量为C
剩余糖果为
$$remaining=C-\sum _{k=0}^{complete}p=C-\sum _{k=0}^{complete}\frac{p(p-1)}{2}$$
且我们知道
$$0\le remaining<complete+1$$
化简得
$$\sqrt{2C+\frac{1}{4}}-\frac{2}{3}<complete\le \sqrt{2C+\frac{1}{4}}-\frac{1}{2}$$
即
$$complete=\lfloor \sqrt{2C+\frac{1}{4}}-\frac{1}{2}\rfloor$$
将complete份完整的礼物分发给 N 个人，共可以分发的回合数：rows = complete / N。
在rows个完整回合中，第 i 个人获得礼物
$$ret[i]=i+(i+N)+(i+2N)+(i+(rows-1)N)=i\times rows+N\frac{rows(rows-1)}{2}$$
在最后一个回合中，可以计算出收到完整礼物的人数 cols = p % N
他们得到的
$$ret[i]+=i+N\times rows$$
最后一名
$$ret[cols]+=remaining$$
最后返回ret即可

class Solution {
public:
    vector<int> distributeCandies(int candies, int num_people) {
        vector<int>ret(num_people, 0);
        int complete = (int)(sqrt(2 * candies + 0.25) - 0.5);
        int remaining = (int)(candies - (complete + 1) * complete * 0.5);
        int rows = complete / num_people;
        int cols = complete % num_people;

        for (int i = 0; i < num_people; i++) {
            ret[i] = (i + 1) * rows + (int)(rows * (rows - 1) * 0.5) * num_people;
            if (i < cols) {
                ret[i] += i + 1 + rows * num_people;
            }
        }

        ret[cols] += remaining;

        return ret;
    }
};