Problem: 2028. 找出缺失的观测数据
Code
class Solution {
public:
vector<int> missingRolls(vector<int>& rolls, int mean, int n) {
vector<int> res;
int s = 0;
int len = rolls.size();
for(auto x: rolls) {
s += x;
}
int m = (mean * (n + len) - s) / n;
int r = (mean * (n + len) - s) % n;
if(m == 6 && r > 0 || m <= 0 || m > 6) return res;
for(int i = 0; i < n ; i ++) {
res.push_back(m);
if(r >= 6 - res[i] && r > 0) {
r = r - (6 - res[i]);
res[i] = 6;
}
else if(r < 6 - res[i] && r > 0) {
res[i] += r;
r = 0;
}
}
return res;
}
};