华为机试真题--最大N个数与最小N个数的和

题目描述：

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

说明：
1. 数组中数字范围[0, 1000]
2. 最大 N 个数与最小 N 个数不能有重叠，如有重叠，输入非法返回-1
3. 输入非法返回-1

输入描述：

第一行输入 M， M 标识数组大小
第二行输入 M 个数，标识数组内容
第三行输入 N，N 表达需要计算的最大、最小 N 个数

输出描述：

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

示例1：

输入：
5
95 88 83 64 100
2

输出：
342
说明：最大 2 个数[100,95],最小 2 个数[83,64], 输出为 342

示例2：

输入：
5
3 2 3 4 2
2
输出：
-1
说明：最大 2 个数[4,3],最小 2 个数[3,2], 有重叠输出为-1

C++源码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_set>
using namespace std;

int specialEncryption(vector<int>& nums, int N) {
    if (N > nums.size() / 2) return -1; // 防止N过大导致重叠

    unordered_set<int> uniqueNums(nums.begin(), nums.end()); // 去重
    if (uniqueNums.size() < nums.size()) return -1; // 有重复元素，输入非法

    vector<int> sortedNums(uniqueNums.begin(), uniqueNums.end());
    sort(sortedNums.begin(), sortedNums.end());

    int sumMax = 0;
    for (int i = sortedNums.size() - N; i < sortedNums.size(); ++i) {
        sumMax += sortedNums[i];
    }

    int sumMin = 0;
    for (int i = 0; i < N; ++i) {
        sumMin += sortedNums[i];
    }

    if (sortedNums.size() - N < N && sortedNums[sortedNums.size() - N] == sortedNums[N - 1]) {
        return -1; // 最大N个数与最小N个数有重叠
    }

    return sumMax + sumMin;
}

int main() {
    int M;
    cin >> M;

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

    int N;
    cin >> N;

    int result = specialEncryption(nums, N);
    cout << result << endl;

    system("pause");
    return 0;
}