LeetCode1619.删除某些元素后的数组均值（Java）

题目

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

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

输入：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

分析

我当时还以为得自己处理结果的格式，结果发现，根本不需要我瞎操心！

法一：暴力解

（凡是暴力解，都是我做的方法，其余方法都是学习别人的思想）
求出每个元素出现的次数，然后一头一尾，慢慢减1，知道删掉这么多元素后，遍历count数组，进行累加，最后返回平均值。
我怀疑我是计数排序写多了，一来就是它。

static public double trimMean(int[] arr) {
		int[] count = new int[100000];
		for (int i = 0; i < arr.length; i++) {
			count[arr[i]]++;
		}
		int len = (int) (arr.length * 0.05);
		int divisor = arr.length - len * 2;
		while (len > 0) {
			for (int i = 0; i < count.length; i++) {
				if (count[i] > 0) {
					count[i]--;
					break;
				}
			}
			for (int j = count.length - 1; j >= 0; j--) {
				if (count[j] > 0) {
					count[j]--;
					break;
				}
			}
			len--;
		}
		double sum = 0;
		for (int i = 0; i < count.length; i++) {
			if (count[i] > 0) {
				sum += (i * count[i]);
			}
		}
		return sum / divisor;
	}

法二（取巧）

排个序，求出前后各要删多少个的长度len，然后去掉数组前len个和数组后len个

static public double trimMean2(int[] arr) {
		// 求出各要删多少个
		int len = (int) (arr.length * 0.05);
		Arrays.sort(arr);
		double sum = 0;
		for (int i = len; i < arr.length - len; i++) {
			sum += arr[i];
		}
		return sum / (arr.length - 2 * len);
	}

后续

打开源码，我觉得很有必要研究一番。

源码

class Solution {
   public double trimMean(int[] arr) {
   //上面的代码
	}
}

public class MainClass {
    public static int[] stringToIntegerArray(String input) {
        input = input.trim();
        input = input.substring(1, input.length() - 1);
        if (input.length() == 0) {
          return new int[0];
        }
    
        String[] parts = input.split(",");
        int[] output = new int[parts.length];
        for(int index = 0; index < parts.length; index++) {
            String part = parts[index].trim();
            output[index] = Integer.parseInt(part);
        }
        return output;
    }
    
    public static String doubleToString(String input) {
        return new DecimalFormat("0.00000").format(input);
    }
    
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String line;
        while ((line = in.readLine()) != null) {
            int[] arr = stringToIntegerArray(line);
            
            double ret = new Solution().trimMean(arr);
            
            String out = doubleToString(ret);
            
            System.out.print(out);
        }
    }
}

感觉源码每一步都是我不太熟练的操作
我得好好学习~
重点想学：

  public static String doubleToString(String input) {
        return new DecimalFormat("0.00000").format(input);
    }

补充一个：
求最大值和最小值，这道题不适用，因为题目是想让你删掉较小数中的5%和较大数的5%

int minVal = Arrays.stream(arr).min().getAsInt();
int maxVal = Arrays.stream(arr).max().getAsInt();

反思

这脑子不行呀，要多多学习，多多思考！！！
end.