力扣刷题记录
模拟
2028. 找出缺失的观测数据
思路
求得n个数得和,再平均分给n个数就可以了。
代码
class Solution {
public int[] missingRolls(int[] rolls, int mean, int n) {
int m = rolls.length;
int sum = 0;
for(int i=0;i<m;i++){
sum += rolls[i];
}
// n 个 数字的和
int nsum = mean * (m + n) - sum;
// 如果和大于6*n 那n个数肯定放不下
// 如果和小于 n 那么可能有一些数等于0 每个数要保证 在1 到 6之间
if( nsum > 6 * n || nsum < n){
return new int[0];
}
int res[] = new int[n];
int yu = nsum % n;
int div = nsum / n;
// 根据余数和商来 赋值给结果数组
for(int i=0;i<n;i++){
res[i] = div + ( (i < yu ) ? 1 : 0 );
}
// 将每个1均匀放入结果数组里
// for(int i=0;i<nsum;i++){
// res[i%n] ++;
// }
return res;
}
}
时间复杂度:O(n)
空间复杂度:O(n)