2028. 找出缺失的观测数据

最新推荐文章于 2024-06-01 00:15:00 发布

原创最新推荐文章于 2024-06-01 00:15:00 发布 · 742 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #找出缺失的观测数据 #构造题

leetcode 专栏收录该内容

89 篇文章

订阅专栏

本文解析了一道力扣上的构造题目，目标是根据已知数组和平均值构造一个新数组，确保其元素值位于1到6之间。文章详细介绍了如何避免不可能情况，并提供了一个高效的构造算法实现。

题目链接：力扣

思路：简单构造题。sumN = n数组的和，sumM = m数组的和。这个题就是构造sumN = (n+m)*mean-sumM 这个数。通过n个数的和构造词出sumN这个数，限制条件就是这n个数的取值范围[1,6]。这不就很简单了。首先考虑不可能的情况。

考虑两个极限值：

1. sumN > n * 6 说明n个数全是最大值6也不行

2. sumN < n 说明n个数全是最小值1也不行

除了这两种情况都是可行的情况。

可行的情况怎么构造呢？我这里考虑通过6和1来构造，可以想到只有一个数可能不是6和1（看代码会懂的）。从一个6开始构造，直到不能放6了，最后剩下的都放1（除了那个可能不是6的数）。详情看代码。

上代码：

class Solution {
    fun missingRolls(rolls: IntArray, mean: Int, n: Int): IntArray {
        val m = rolls.size
        val sum = (n + m) * mean
        var sumM = 0
        for (roll in rolls) {
            sumM += roll
        }
        val sumN = sum - sumM
        if (sumN > n * 6 || sumN < n) {
            return IntArray(0)
        }
        val result = IntArray(n) { 1 }
        for (i in 0 until n) {
            var sumTmp = n - i - 1 + 6 * (i + 1)
            if (sumTmp < sumN) {
                result[i] = 6
            } else if (sumTmp == sumN) {
                result[i] = 6
                break
            } else {
                var x = 6
                while (sumTmp > sumN) {
                    sumTmp--
                    x--
                }
                result[i] = x
                break
            }
        }
        return result
    }
}