1619. 删除某些元素后的数组均值

1619. 删除某些元素后的数组均值

给你一个整数数组 arr ,请你删除最小 5% 的数字和最大 5% 的数字后,剩余数字的平均值。

标准答案 误差在 10-5 的结果都被视为正确结果。

 

示例 1:

输入:arr = [1,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,3]
输出:2.00000
解释:删除数组中最大和最小的元素后,所有元素都等于 2,所以平均值为 2 。

示例 2:

输入:arr = [6,2,7,5,1,2,0,3,10,2,5,0,5,5,0,8,7,6,8,0]
输出:4.00000

示例 3:

输入:arr = [6,0,7,0,7,5,7,8,3,4,0,7,8,1,6,8,1,1,2,4,8,1,9,5,4,3,8,5,10,8,6,6,1,0,6,10,8,2,3,4]
输出:4.77778
func trimMean(arr []int) float64 {
    // 题目中说明长度为20的倍数
    nums := len(arr) / 20

    sort.Slice(arr, func(i, j int) bool {return arr[i] < arr[j]})

    var res float64
    for i,_ := range arr {
        if i < nums || i >= (len(arr) - nums) {
            continue
        }
        res += float64(arr[i])
    }
    return res/float64((len(arr) - 2*nums))
}
// 官方解法
func trimMean(arr []int) float64 {
    sort.Ints(arr)
    n := len(arr)
    sum := 0
    for _, x := range arr[n/20 : 19*n/20] {
        sum += x
    }
    return float64(sum*10) / float64(n*9)
}

 思路:排序后掐头去尾

posted on 2022-09-14 10:47  HHHuskie  阅读(20)  评论(0编辑  收藏  举报

导航