思路:1-6之间的的n个数组合起来要变成sum_t = mean*(rolls.size()+n) - sum(rolls) ;
那么可以先假设每个数都是sum_t / n 其中这个数必须要在1 - 6 之间否者无法分配。
然后可以得出n * (sum_t / n ) <= sum ; 需要对余数mod进行调整,为了减少调整的次数,每次尽量向上调整;最终调整完之后 mod 必须为0;
code:
class Solution {
public:vector<int> missingRolls(vector<int>& rolls, int mean, int n) {int len = rolls.size();int sum = (n + len) * mean;for(int i = 0 ; i < len; i++){sum -= rolls[i];}//从1-6找出n个数,求和等于sumif(sum / n > 6 || sum / n < 1)return {};//为答案数组分配结果, n * (sum / n) <= sum , vector<int> ans(n, sum / n);int mod = sum % n; //多余的部分,对答案进行调整,如果ans[i]小于6就可以进行一次调整,直到消除这个modint i = 0;while(mod && i < n){if(ans[i] < 6){if(6 - ans[i] >= mod){ans[i] += mod;mod = 0;}else{mod -= (6 - ans[i]);ans[i] = 6;}}i++;}if(mod != 0)return {};return ans;}
};