力扣题解——不同的平均值数目

题目描述

题目链接在此：不同的平均值数目

给你一个下标从 0 开始长度为 偶数 的整数数组 nums 。

只要 nums 不是 空数组，你就重复执行以下步骤：

找到 nums 中的最小值，并删除它。
找到 nums 中的最大值，并删除它。
计算删除两数的平均值。
两数 a 和 b 的 平均值 为 (a + b) / 2 。

比方说，2 和 3 的平均值是 (2 + 3) / 2 = 2.5 。
返回上述过程能得到的 不同 平均值的数目。

注意 ，如果最小值或者最大值有重复元素，可以删除任意一个。

样例1

输入：nums = [4,1,4,0,3,5]
输出：2
解释：
1. 删除 0 和 5 ，平均值是 (0 + 5) / 2 = 2.5 ，现在 nums = [4,1,4,3] 。
2. 删除 1 和 4 ，平均值是 (1 + 4) / 2 = 2.5 ，现在 nums = [4,3] 。
3. 删除 3 和 4 ，平均值是 (3 + 4) / 2 = 3.5 。
2.5 ，2.5 和 3.5 之中总共有 2 个不同的数，我们返回 2 。

样例2

输入：nums = [1,100]
输出：1
解释：
删除 1 和 100 后只有一个平均值，所以我们返回 1 。

分析

题目要求找到不同的平均值，那平均值相同则证明原数相同，所以这里我们可以直接进行排序，然后将首位元素与末尾元素相加，利用set集合的特性进行去重，最后返回set集合的大小就可以了。

AC代码

class Solution {
public:
    int distinctAverages(vector<int>& nums) {
        // 初始化set集合
        set<int> result;
        // 对nums进行排序，保证首位元素最小，末尾元素最大
        sort(nums.begin(),nums.end());
        // 循环nums直至没有元素
        while(!nums.empty()){
            // 利用set集合特性，将两数之和存入集合，保证唯一性
            result.insert(nums.front()+nums.back());
            // 删除首位元素
            nums.erase(nums.begin());
            // 删除末尾元素
            nums.pop_back();
        }
        // 返回set集合大小
        return result.size();

    }
};