Java&C++题解与拓展——leetcode2028.找出缺失的观测数据【vector学习与使用】

每日一题做题记录，参考官方和三叶的题解

题目要求

思路

可以算出丢失的$n$个值之和$sum$，然后构造一个加起来恰好得$sum$的数组res即可【${\sum }_{i=0}^{n}res[i]=sum$】。

• $res[i]\in [1,6]$，所以$sum\in [n,6\ast n]$，否则无解返回空数组；
• 有解的时候，一定可以构造出一个包含$a$个$\lfloor \frac{sum}{n}\rfloor$和$n-a$个$\lfloor \frac{sum}{n}\rfloor +1$的结果。那么将res填充为$\lfloor \frac{sum}{n}\rfloor$，然后计算需要补$1$的数量，也就是$sum$与$\lfloor \frac{sum}{n}\rfloor \ast n$的差值$dis$，将这些$1$加在res的$dis$个数上。

Java

class Solution {
    public int[] missingRolls(int[] rolls, int mean, int n) {
        int m = rolls.length, cnt = m + n;
        int sum = mean * cnt;
        for(int i : rolls)
            sum -= i;
        if(sum < n || sum > 6 * n)
            return new int[0];
        int[] res = new int[n];
        //构造结果
        Arrays.fill(res, sum / n);
        //保证整除
        if(sum / n * n < sum) {
            int dis = sum - (sum / n * n), idx = 0;
            while(dis-- > 0)
                res[idx++]++;
        }
        return res;
    }
}

• 时间复杂度：$O(m+n)$
• 空间复杂度：$O(n)$

C++

class Solution {
public:
    vector<int> missingRolls(vector<int>& rolls, int mean, int n) {
        int m = rolls.size(), cnt = m + n;
        int sum = mean * cnt;
        for(int i : rolls)
            sum -= i;
        if(sum < n || sum > 6 * n)
            return {};
        //构造结果
        vector<int> res(n, sum / n);
        //保证整除
        if(sum / n * n < sum) {
            int dis = sum - (sum / n * n), idx = 0;
            while(dis-- > 0)
                res[idx++]++;
        }
        return res;
    }
};

• 时间复杂度：$O(m+n)$
• 空间复杂度：$O(n)$

vector

• 学习参考链接
• 前两天刚学习整理过，今天主要是学习一下如何快速插入多个相同的元素。
• 如何添加$cnt$个$val$，参考
  • 初始化时直接添加：

  std::vector<int> res(cnt, val);
  

  • 用insert函数

  res.insert(res.end(), cnt, val);//在末尾添加
  

总结

题目思路不难，主要是想好如何构造能整除的答案。

欢迎指正与讨论！