【哈希集合】最大N个数与最小N个数的和

题目描述与示例

题目描述

给定一个数组，编写一个函数来计算它的最大N个数与最小N个数的和。你需要对数组进行去重。

输入描述

第一行输入M， M标识数组大小

第二行输入M个数，标识数组内容

第三行输入N，N表达需要计算的最大、最小N个数

输出描述

输出最大N个数与最小N个数的和。

补充说明

数组中数字范围[0，1000]

最大N个数与最小N个数不能有重叠，如有重叠返回-1

示例

输入

5 95 88 83 64 100 2

输出

342

解题思路

直白、简单的题目。主要考察以下几个知识点

1. 哈希集合去重

2. 去重后的结果再次排序得到新列表lst_new

3. lst_new的长度必须大于2*N

4. 计算最大的N个数和最小的N个数的和

代码

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>

using namespace std;

int main() {
    int m, n;
    cin >> m;

    vector<int> lst(m);
    for (int i = 0; i < m; ++i) {
        cin >> lst[i];
    }

    cin >> n;

    // 使用 set 进行去重，并对其进行排序
    set<int> uniqueSet(lst.begin(), lst.end());//使用 set 容器存储的各个元素的值必须各不相同
    vector<int> lstNew(uniqueSet.begin(), uniqueSet.end());
    sort(lstNew.begin(), lstNew.end());

    if (lstNew.size() < 2 * n) {
        cout << -1 << endl;
    }
    else {
        int sum = 0;
        for (int i = 0; i < n; ++i) {
            sum += lstNew[i] + lstNew[lstNew.size() - 1 - i];
        }
        cout << sum << endl;
    }

    return 0;
}

时空复杂度

时间复杂度：O(MlogM)。排序所需时间复杂度

空间复杂度：O(M)